Expert-Guided Explainable Few-Shot Learning for Medical Image Diagnosis

#### Контекст Медицинский анализ изображений широко применяется в диагностике, но сталкивается с ограничениями, связанными с недостатком экспертно аннотированных данных. Это приводит к проблемам с общеупотребимостью моделей и их клиническим применением. Большая часть работы в области небольших выборок (few-shot learning) ориентирована на улучшение точности классификации, но часто игнорирует важность прозрачности и интерпретируемости моделей, которые критичны для принятия решений в медицине. Наша мотивация заключается в создании фреймворка, который бы не только повышал точность, но и обеспечивал понятные причины своих выводов, укрепляя доверие клинических специалистов. #### Метод Мы предлагаем фреймворк, основанный на нескольких компонентов. Во-первых, мы используем регионы интереса (ROIs), предоставляемые радиологами, для направления внимания модели на ключевые области изображений. Во-вторых, мы вводим экспертно руководствованный метод управления вниманием, основанный на Grad-CAM, для получения специфичных для задачи специфичных супервайзеров внимания. В-третьих, мы предлагаем новую функцию потерь, основанную на Dice-similarity, для выравнивания внимания модели с теми областями, которые клинически значимы. Эта функция потерь учитывается вместе с целевым классическим целевым функционированием в рамках нескольких выборок. Модель обучается с использованием прототипной сети, при этом данные обучения сокращаются до нескольких выборок для каждого класса. #### Результаты Мы проверили наш фреймворк на двух различных наборах данных: BraTS (МРТ) и VinDr-CXR (Chest X-ray). На BraTS, в базовой ситуации точность модели выросла с 77.09% до 83.61%, а на VinDr-CXR — с 54.33% до 73.29%. Эти улучшения показывают, что наше решение повышает качество классификации и улучшает интерпретируемость модели. Мы также предоставили Grad-CAM-визуализации, подтверждающие, что модель обучается сосредоточиваться на диагностически важных областях, что улучшает надежность и надежность решений. #### Значимость Наш фреймворк может быть применен в различных областях, где недостаточно данных для обучения моделей, но требуется высокая точность и прозрачность. Это включает диагностику болезней, анализ изображений, где недостаточно выборок для обучения, и другие проблемы, где необходима доверительная взаимоотношение с клиническими специалистами. Наш подход предоставляет значительное преимущество в скорости интерпретируемости, что помогает клиническим специалистам понимать исходные решения и увеличивает ценность моделей в реальной клинической практике. #### Выводы Мы разработали экспертно руководствованную сист