Explainable AI Technique in Lung Cancer Detection Using Convolutional Neural Networks

2508.10196v1 eess.IV, cs.CV, 68T07 2025-08-15
Авторы:

Nishan Rai, Sujan Khatri, Devendra Risal

Резюме на русском

## Контекст Определение рака легких на ранней стадии является критически важным для улучшения прогноза болезни и повышения выживаемости пациентов. Однако традиционные методы диагностики часто требуют значительных временных и ресурсных затрат, а также зависят от опыта специалистов. Это приводит к задержкам в обнаружении и несогласованности в диагностике. Нейронные сети, особенно глубокие, показали свою эффективность в автоматизации такой диагностики. Несмотря на это, многие из них не предоставляют понятных объяснений для клинических специалистов, что ограничивает их применение в практических ситуациях. ## Метод Мы применили экзордическую архитектуру сверточных нейронных сетей (CNN), настроенную на диагностику рака легких с помощью трех классов: нормальное состояние, бенигная форма рака и малянгная форма. Использовались данные из набора IQ-OTH/NCCD (1,197 сканов). Для обучения использовалась пользовательская CNN-архитектура и три предварительно обученных модели: DenseNet121, ResNet152 и VGG19. Для борьбы с неравномерностью классов применялась техника cost-sensitive learning. Для улучшения прозрачности интерпретируемости результатов внедряется метод Shapley Additive Explanations (SHAP). ## Результаты Модели DenseNet121 и ResNet152 показали самые высокие показатели в сравнении с другими моделями. DenseNet121 демонстрировал лучшую сбалансированность в показателях precision (92%), recall (90%) и F1-score (91%). ResNet152, в свою очередь, достиг более высокой точности (97.3%). Эти модели были оценены с использованием метрик ROC-AUC, accuracy и других, что подтвердило их высокую эффективность в диагностике. Дополнительно, SHAP-анализ позволил показать, какие части изображений в CT-сканах способствуют данному прогнозу, что улучшило понимание решений модели. ## Значимость Результаты имеют большое значение для клинической практики, так как обеспечивают быструю, точную и понятную поддержку в диагностике рака легких. Это особенно важно для ресурсораспределенных сред, где доступ к экспертам ограничен. Интеграция explainability в модели не только повышает доверие к результатам, но и позволяет клиническим специалистам более эффективно использовать информацию в решениях о лечении. ## Выводы Результаты исследований показали, что CNN-модели с интегрированной explainability могут предоставить эффективную и надежную поддержку в диагностике рака легких. Область применения таких моделей включает не только клинические ситуации, но и мониторинг здоровья населения в условиях удаленного контроля. Будущие исследования будут посвящены улучшению точности моделей, расширению наборов д

Abstract

Early detection of lung cancer is critical to improving survival outcomes. We present a deep learning framework for automated lung cancer screening from chest computed tomography (CT) images with integrated explainability. Using the IQ-OTH/NCCD dataset (1,197 scans across Normal, Benign, and Malignant classes), we evaluate a custom convolutional neural network (CNN) and three fine-tuned transfer learning backbones: DenseNet121, ResNet152, and VGG19. Models are trained with cost-sensitive learning to mitigate class imbalance and evaluated via accuracy, precision, recall, F1-score, and ROC-AUC. While ResNet152 achieved the highest accuracy (97.3%), DenseNet121 provided the best overall balance in precision, recall, and F1 (up to 92%, 90%, 91%, respectively). We further apply Shapley Additive Explanations (SHAP) to visualize evidence contributing to predictions, improving clinical transparency. Results indicate that CNN-based approaches augmented with explainability can provide fast, accurate, and interpretable support for lung cancer screening, particularly in resource-limited settings.

Ссылки и действия