Enhanced Liver Tumor Detection in CT Images Using 3D U-Net and Bat Algorithm for Hyperparameter Optimization
2508.08452v1
cs.LG, cs.CV
2025-08-14
Авторы:
Nastaran Ghorbani, Bitasadat Jamshidi, Mohsen Rostamy-Malkhalifeh
Резюме на русском
## Контекст
Одним из наиболее распространенных и смертельных видов рака является злокачественный новообразование печени. Раннее выявление и точная диагностика необходимы для эффективного лечения этой заболевания. Однако диагностика злокачественных опухолей печени чрезвычайно сложная, ввиду того, что ткани печени имеют очень мало контрастности и требуют очень точных методов выделения интересующих областей на изображениях. Традиционные методы диагностики часто требуют значительного времени и специализированных навыков, что может привести к неточности в результатах. Автоматизированные системы, основанные на искусственном интеллекте, могут существенно повысить точность диагностики, уменьшить время работы специалиста и повысить общую эффективность лечения. Однако существуют ряд проблем, связанных с точностью и производительностью этих систем, в частности, трудности с оптимизацией параметров нейронных сетей.
## Метод
В данной работе предлагается использовать сверточную нейронную сеть типа 3D U-Net для выделения интересующих областей на изображениях. 3D U-Net широко используется в задачах медицинского изображения для выделения интересующих областей в трехмерных изображениях. Однако улучшение точности и результатов зачастую зависит от того, настроены ли параметры модели правильно. Для этого в подходе используется метод оптимизации Bat Algorithm. Bat Algorithm является мета-хиерархическим методом оптимизации, который моделирует поведение летучих мышей при поиске лучшего решения по полям функции целевой функции. Этот метод оптимизирует ключевые параметры модели, такие как learning rate и batch size, для повышения точности и надежности модели.
## Результаты
Запуск модели проводился на общедоступном наборе данных, который содержит изображения CT с метками, обозначающими зоны подозрительных областей. Модель была оптимизирована с помощью Bat Algorithm, что позволило повысить приверженность модели к точности и уменьшить возможность выявления ложных срабатываний. Результаты экспериментов показали, что модель способна достигать высоких результатов в области F1-меры, даже при низких предсказательных порогов. Это крайне важно для диагностики, где необходимо избегать пропуска подозрительных областей, так как это может привести к задержке лечения. Таким образом, полученные результаты свидетельствуют о том, что интеграция метода 3D U-Net и Bat Algorithm может стать мощным инструментом для автоматизации диагностики злокачественных новообразований печени.
## Значимость
Предлагаемый подход может быть применен в различных областях клинической практики, в том числе в диагностике злокачественных состояний печени. Это
Abstract
Liver cancer is one of the most prevalent and lethal forms of cancer, making
early detection crucial for effective treatment. This paper introduces a novel
approach for automated liver tumor segmentation in computed tomography (CT)
images by integrating a 3D U-Net architecture with the Bat Algorithm for
hyperparameter optimization. The method enhances segmentation accuracy and
robustness by intelligently optimizing key parameters like the learning rate
and batch size. Evaluated on a publicly available dataset, our model
demonstrates a strong ability to balance precision and recall, with a high
F1-score at lower prediction thresholds. This is particularly valuable for
clinical diagnostics, where ensuring no potential tumors are missed is
paramount. Our work contributes to the field of medical image analysis by
demonstrating that the synergy between a robust deep learning architecture and
a metaheuristic optimization algorithm can yield a highly effective solution
for complex segmentation tasks.
Ссылки и действия
Дополнительные ресурсы: