Toward a robust lesion detection model in breast DCE-MRI: adapting foundation models to high-risk women

## Контекст Малоинвазивные методы диагностики рака молочной железы, такие как ДКЭ-МРТ, играют ключевую роль в раннем выявлении онкологических заболеваний. Однако обеспечение высокой точности в данном контексте связано с рядом проблем. Несбалансированные и разнообразные клинические данные, а также необходимость интерпретируемых и надежных моделей делают задачу выявления неоднородных объектов сложной. Имеющиеся подходы часто страдают от отсутствия общей гибкости и недостаточной подробности в анализе. Модели требуются, которые могут обрабатывать большие объемы данных, поддерживать интерпретируемость и обеспечивать высокую точность в разных условиях. ## Метод Предложена архитектура сверточного пайплайна, объединяющая эффективные технологии основных моделей и гибкие способы классификации. Базовой моделью стала Medical Slice Transformer (MST), обученная с использованием DINOv2 для самостоятельного обучения. MST генерирует высококачественные описательные признаки для каждого срезов магнитно-резонансной импеданси. Эти признаки подаются на вход в Kolmogorov--Arnold Network (KAN), которая использует адаптивные B-сплайны для локализованного преобразования признаков. Это дает модели гибкость в обработке сложностей неоднородных данных. Результаты обучения используются для создания интерпретируемых тепловых карт, позволяющих визуализировать участки входных изображений, которые были ключевыми для принятия решений. ## Результаты Использованы ДКЭ-МРТ данные, полученные в рамках клинических исследований. Модель MST+KAN протестирована на выборках с разным уровнем неравенства классов и разнообразием клинических данных. Полученные результаты показали, что модель достигает AUC = 0.80 ± 0.02 при классификации бенинных и малигонных образов. Это существенно превышает результаты базовой модели MST. Также были получены тепловые карты, которые акцентируют внимание на важных частях изображений, повышая прозрачность модели. ## Значимость Предложенный подход может быть применен в различных ситуациях диагностики, где требуется раннее и точное выявление рака молочной железы. Он предоставляет значительные преимущества, включая высокую точность и интерпретируемость, что является критически важной особенностью для клинического применения. Этот подход может способствовать улучшению качества здравоохранения, особенно для высокорисковой группы пациенток, обеспечивая доступность и надежность диагностических средств. ## Выводы Модель MST+KAN доказала свою эффективность в области выявления неоднородных и сильно небалансированных клинических д