RUS  ENG
Полная версия
ЖУРНАЛЫ // Искусственный интеллект и принятие решений // Архив

Искусственный интеллект и принятие решений, 2024, выпуск 2, страницы 116–122 (Mi iipr592)

Машинное обучение, нейронные сети

Разработка трёхмерной сверточной нейронной сети с вниманием для обнаружения аневризм

С. Г. Синицаa, Е. И. Зябловаb, Д. О. Кардаильскаяb, И. А. Заяцa, А. А. Халафянa, А. В. Ищенкоc

a Кубанский государственный университет, Краснодар, Россия
b ГБУЗ "Научно-исследовательский институт – Краевая клиническая больница 1 имени С. В. Очаповского" Mинистерства здравоохранения Краснодарского края, Краснодар, Россия
c ООО "КУБ", Краснодар, Россия

Аннотация: Рассмотрен прототип трехмерной сверточной нейронной сети с блоком внимания, определяющей вероятность наличия признаков интракраниальных аневризм головного мозга в отдельном исследовании контрастной компьютерной томографии-ангиографии. Для обучения сети использовались данные контрастных компьютерной томографии-ангиографии в формате DICOM с содержанием интракраниальных аневризм и без них. Данные были разделены на тренировочное и валидационное подмножества в пропорции 65 и 35%, соответственно. С применением библиотек Keras и Tensorflow в среде программирования Python разработана модель трехмерной сверточной нейронной сети со входными данными размерности 192$\times$192$\times$128, состоящая из четырех слоев свертки с ядром размерности 3 и блока внимания на себя. Меткость, точность и полнота классификации на тестовой выборке достигли 96, 99 и 93%, соответственно, что превысило показатели известных ранее нейросетей.

Ключевые слова: компьютерная томография, ангиография, интракраниальные аневризмы, DICOM, машинное обучение, трехмерная сверточная нейронная сеть, внимание.

DOI: 10.14357/20718594240209



Реферативные базы данных:


© МИАН, 2024