$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Drie vooraf getrainde deep learning-modellen, namelijk MobileNetV2, EfficientNetV1 en Inception-ResNet-V2, werden geëvalueerd voor vier-klasse hersentumorclassificatie met behulp van MRI-beelden.
Gebruikte dataset
Het voorgestelde framework werd geëvalueerd met behulp van een openbaar beschikbare hersentumor MRI-dataset verkregen van de Kaggle-platform. De dataset bestond uit 835 normale hersen-MRI-beelden, 826 glioma-beelden, 822 meningeoom-beelden en 827 hypofysetumor-beelden. Representatieve MRI-voorbeelden uit de dataset worden getoond in Figuur 2.
MobileNetV2
Het MobileNetV2-model bevatte 2.263.108 totale parameters, waaronder 2.228.996 trainbare en 34.112 niet-trainbare. De classificatieprestatiemetrieken worden gepresenteerd in Tabel 2. Het model behaalde een algemene classificatie-nauwkeurigheid van 97%, met vergelijkbare macro- en gewogen gemiddelde F1 Scores. De verwarringsmatrix die wordt getoond in Tabel 3 toont een sterke classificatieprestatie voor alle vier tumorcategorieën, met relatief weinig verkeerde classificaties tussen glioma en meningeoom.
Figuur 3 toont de trainings- en validatie-nauwkeurigheid/verliescurves voor MobileNetV2. Het model vertoonde stabiele convergentie na verschillende trainingsepochen, waarbij de validatieprestaties geleidelijk verbeterden.
Inception-ResNet-V2
Het Inception-ResNet-V2 model had 54.342.884 totale parameters, waaronder 54.282.340 trainbare en 60.544 niet-trainbare. De classificatiemetrieken worden samengevat in Tabel 4. Het model behaalde een algemene classificatie-nauwkeurigheid van 98%, wat een verbeterde prestatie aantoont in vergelijking met MobileNetV2.
De verwarringsmatrix in Tabel 5 geeft een hoge classificatie-nauwkeurigheid aan voor alle tumorcategorieën met verminderde foute classificaties. Er werd een betere discriminatie tussen tumorklassen waargenomen in vergelijking met het MobileNetV2-model. Figuur 4 illustreert de trainings- en validatieprestatiecurven voor Inception-ResNet-V2. Het model vertoonde stabiel leergedrag en consistente convergentie gedurende de trainingsepochen.
EfficientNetV1
Het EfficientNetV1-model bevatte 6.580.363 totale parameters, waaronder 6.518.308 trainbare en 62.055 niet-trainbare. De classificatiemetrieken die worden samengevat in Tabel 6 tonen de hoogste prestatie onder de geëvalueerde modellen, met een nauwkeurigheid, precisie, recall en F1-score van ongeveer 99%.
De verwarringsmatrix in Tabel 7 toont een sterke classificatieprestatie voor alle vier tumorklassen, met minimale verkeerde classificaties. Het model classificeerde de meeste glioma-, meningeoom-, geen-tumor- en hypofysetumorbeelden correct.
Figuur 5 toont de trainings- en validatie-nauwkeurigheid/verliescurves voor EfficientNetV1. Het model behaalde stabiele convergentie met lagere validatieschommelingen in vergelijking met de andere geëvalueerde architectuur.
Algemene vergelijking
Van de geëvalueerde modellen behaalde EfficientNetV1 de hoogste classificatieprestatie terwijl het een lagere computationele complexiteit behield dan Inception-ResNet-V2. MobileNetV2 toonde efficiënte lichtgewichtprestaties met verminderde parametervereisten, terwijl Inception-ResNet-V2 een verbeterde classificatie-nauwkeurigheid behaalde ten koste van aanzienlijk hogere modelcomplexiteit. De bevindingen geven aan dat EfficientNetV1 de beste balans biedt tussen classificatie-nauwkeurigheid en computationele efficiëntie voor vier-klasse hersentumor MRI-classificatie.
Beschikbaarheid van gegevens
De hersen-MRI-dataset die in dit onderzoek werd geanalyseerd, is openbaar beschikbaar via het Kaggle-platform. Alle gegevens die zijn gegenereerd en geanalyseerd tijdens dit onderzoek, inclusief classificatieresultaten, prestatiemetrieken, verwarringsmatrices en trainings-/validatieresultaten, worden gepresenteerd in de figuren en tabellen van dit artikel. De aangepaste code die is gebruikt voor modelontwikkeling en evaluatie, is beschikbaar bij de corresponderende auteur op redelijk verzoek.

Figuur 1: Aandachtsgestuurde transfer learning-framework voor vier-klasse hersentumorclassificatie. De workflow illustreert preprocessing, aandachtsgebaseerde functie-extractie, convolutionele functie-learning, functiefusie en volledig verbonden classificatielagen die worden gebruikt voor MRI-gebaseerde tumorclassificatie. Klik hier om een grotere versie van deze figuur te bekijken.

Figuur 2: Representatieve MRI-beelden van de vier-klasse hersentumor dataset.
Voorbeeld MRI-scans die overeenkomen met glioma, meningeoom, hypofysetumor en normale hersenklassen die worden gebruikt voor modeltraining en evaluatie worden gepresenteerd. Klik hier om een grotere versie van deze figuur te bekijken.

Figuur 3: Trainings- en validatie-nauwkeurigheid/verliescurves voor MobileNetV2. De grafieken