Uncertainty-Aware Learning Policy for Reliable Pulmonary Nodule Detection on Chest X-Ray

2508.13236v1 eess.IV, cs.AI, cs.CV 2025-08-21
Авторы:

Hyeonjin Choi, Jinse Kim, Dong-yeon Yoo, Ju-sung Sun, Jung-won Lee

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

#### ## Контекст Область диагностики люминального рака лёгкого, одного из наиболее распространённых видов злокачественных новообразований, представляет собой критически важную задачу, так как раннее выявление и агрессивное лечение могут существенно повысить прогноз заболевания. Однако необходимость в надежной и точной диагностике сталкивается с рядом проблем, в том числе относительной степени достоверности диагноста, зависящей от опыта и утомительности врача-диагноста. Несмотря на то, что Медицинский Искусственный Интеллект (МИИ) становится всё более распространённым средством помощи в диагностике, многие врачи остаются скептическими относительно достоверности его выводов, что мешает полному использованию МИИ в клинической практике. Основной причиной этого скептицизма является недостаточное описание неопределённости в выводах МИИ, что может привести к ошибкам диагностики. Для решения этих проблем предлагается разработать подход, учитывающий не только информацию о подозрительных областях на рентгеновских снимках груди, но и фоновую знаний, которую обычно используют врачи в процессе диагностики. #### ## Метод Предложенная модель Uncertainty-Aware Learning Policy основывается на комбинации информации с рентгеновских снимков груди и знаний о классических приёмах диагностики, используемых врачами. Основная идея состоит в том, что не только изображения, но также фоновые знания врачей, такие как клинические опыт и общие принципы диагностики, должны быть включены в обучение модели. Модель использует архитектуру, которая позволяет не только выявлять подозрительные области на изображениях, но также оценивать неопределённость в своих выводах. Эта неопределённость измеряется с помощью метрики энтропии, которая снижается в результате обучения модели. Данные для обучения и тестирования были получены из Аджо Университетского Больничного Издания, включая 2517 изображений без подозрительных образований и 656 изображений с подозрительными образованиями. #### ## Результаты В результате экспериментов показано, что модель Uncertainty-Aware Learning Policy достигла значительных улучшений по сравнению с базовой моделью. Она показала улучшение точности диагностики (детекции областей, содержащих подозрительные образования) на 92% (интервал доверия 0.2 / FPPI 2), что составляет прирост в 10% по сравнению с базовой моделью. Более того, модель уменьшила меру неопределённости (энтропию) на 0.2, что означает более уверенный вывод в предсказаниях. Эти результаты показывают, что включение фонового знания в обучение значительно повышает точность и уменьшает неопреде

Abstract

Early detection and rapid intervention of lung cancer are crucial. Nonetheless, ensuring an accurate diagnosis is challenging, as physicians' ability to interpret chest X-rays varies significantly depending on their experience and degree of fatigue. Although medical AI has been rapidly advancing to assist in diagnosis, physicians' trust in such systems remains limited, preventing widespread clinical adoption. This skepticism fundamentally stems from concerns about its diagnostic uncertainty. In clinical diagnosis, physicians utilize extensive background knowledge and clinical experience. In contrast, medical AI primarily relies on repetitive learning of the target lesion to generate diagnoses based solely on that data. In other words, medical AI does not possess sufficient knowledge to render a diagnosis, leading to diagnostic uncertainty. Thus, this study suggests an Uncertainty-Aware Learning Policy that can address the issue of knowledge deficiency by learning the physicians' background knowledge alongside the Chest X-ray lesion information. We used 2,517 lesion-free images and 656 nodule images, all obtained from Ajou University Hospital. The proposed model attained 92% (IoU 0.2 / FPPI 2) with a 10% enhancement in sensitivity compared to the baseline model while also decreasing entropy as a measure of uncertainty by 0.2.

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