Deze studie omvatte geen werving van menselijke deelnemers of de verzameling van identificeerbare patiëntgegevens. Alle experimenten werden uitgevoerd met behulp van de publiek beschikbare ISIC 2020 huidlaesie-dataset verkregen uit de Kaggle-repository; daarom waren goedkeuring van een institutionele ethische commissie en geïnformeerde toestemming niet vereist. Alle materialen die in deze studie zijn gebruikt, staan in de Tabel van Materialen.
Kenmerkextractie en fusie
Afbeeldingen van huidlaesies werden verwerkt met behulp van meerdere vooraf getrainde CNN-modellen. Diepe kenmerkvectoren, geëxtraheerd uit de geselecteerde vooraf getrainde CNN-architecturen, worden samengevoegd om een uniforme kenmerkrepresentatie voor elke afbeelding van de huidlaesie te construeren. De toegepaste fusiestrategie behoudt complementaire visuele informatie die door verschillende CNN-architecturen is aangeleerd, waardoor de daaropvolgende SVM-classifier zowel laagwaardige textuurkenmerken als hoogwaardige semantische representaties kan benutten. Hoewel dimensiereductietechnieken, zoals principale componentanalyse of op wederzijdse informatie gebaseerde kenmerkselectie, de dimensionaliteit van de kenmerken kunnen verminderen, is de volledige gefuseerde representatie bewust behouden omdat de gefuseerde kenmerkruimte computationeel hanteerbaar blijft voor de gebruikte RBF-SVM-classifier, terwijl de complementaire diagnostische informatie uit de heterogene CNN-backbones behouden blijft.
Classificatie
De gefuseerde kenmerkvectoren werden gebruikt om een SVM-classifier met een RBF-kernel te trainen. Er werden technieken voor hyperparameteroptimalisatie toegepast om de optimale classificatie-instellingen te bepalen. De classifier categoriseerde vervolgens de huidlaesies in hun respectievelijke klassen.
Uitlegbaarheid
Om de interpreteerbaarheid te verbeteren, werd LIME toegepast om visuele verklaringen te genereren die de beeldregio's markeren die het meest significant bijdragen aan de classificatieresultaten. Deze verklaringen zouden clinici helpen bij het begrijpen en valideren van de beslissingen van het model.
Evaluatieplan
Het voorgestelde framework is geëvalueerd met behulp van een benchmarkdataset van huidlaesies. De prestaties zijn beoordeeld op basis van accuratesse, precisie, recall en de F1-score. Er zijn vergelijkende experimenten uitgevoerd ten opzichte van bestaande benaderingen van machine learning en deep learning om de effectiviteit van het voorgestelde framework aan te tonen.
Verwachte resultaten
Van de studie werd verwacht dat deze zou leiden tot een accuraat en interpreteerbaar classificatiesysteem voor huidlaesies. Voorlopige experimentele resultaten tonen aan dat EDLF-SLC een nauwkeurigheid van 88,0%, precisie van 89,0%, recall van 87,0% en een F1-score van 88,0% behaalt, waarmee het meerdere concurrerende methoden overtreft. De integratie van LIME-uitleggen was bedoeld om de betrouwbaarheid te verhogen en de adoptie van AI-ondersteunde diagnostische systemen in de klinische praktijk te vergemakkelijken.
Betekenis
Dit onderzoek draagt bij aan het groeiende veld van uitlegbare AI in de gezondheidszorg door zowel prestatie- als transparantie-uitdagingen bij de diagnose van huidlaesies aan te pakken. Het voorgestelde raamwerk heeft de potentie om dermatologen te ondersteunen bij het nemen van betrouwbaardere en interpreteerbare diagnostische beslissingen, wat uiteindelijk de patiëntzorg verbetert. Bovendien verstrekken de auteurs in deze sectie informatie over de materialen en de methodologie die in deze onderzoeksstudie zijn toegepast. Specifiek beginnen de auteurs met de beschrijving van de dataset die voor de experimenten is gebruikt, gevolgd door een gedetailleerde bespreking van het Explainable Deep Learning Framework voor de classificatie van huidlaesies.
Dataset
Het gepresenteerde werk is gebaseerd op de publiekelijk toegankelijke ISIC 2020-dataset (International Skin Imaging Collaboration), die kan worden geraadpleegd op de Kaggle-website (https://www.kaggle.com/datasets/fanconic/skin-cancer-malignant-vs-benign). De ISIC-repository wordt erkend als een van de beste bronnen op het gebied van dermatologische beeldvorming, aangezien deze een grote variëteit aan dermoscopische beelden van verschillende soorten huidlaesies biedt, zoals weergegeven in Figuur 1. Cruciaal is dat de dataset beelden bevat van zowel benigne als maligne aandoeningen, waardoor deze geschikt is voor het uitvoeren van binaire classificatietaken voor huidkanker 46. De huidige dataset bevat 3.297 beelden, waarvan er 1.800 benigne en 1.497 maligne zijn. Voor efficiëntere experimenten moest de data worden opgesplitst in een trainingsset en een testset. In het bijzonder zijn er 2.637 trainingsbeelden, waaronder 1.440 benigne en 1.197 maligne, terwijl de testset bestaat uit 660 beelden, waarvan 360 benigne en 300 maligne zijn. Een samenvatting van de dataset is weergegeven in Tabel 2.
Het voorgestelde EDLF-SLC-model
In dit artikel is een efficiënte en begrijpelijke oplossing voorgesteld voor het classificeren van huidlaesies in goedaardige en kwaadaardige categorieën, met behulp van een nieuwe uitlegbare deep learning-techniek gebaseerd op een hybride benadering die de voordelen van meerdere vooraf getrainde convolutionele neurale netwerkmodellen combineert. Convolutionele neurale netwerken zijn zeer effectief gebleken in het extraheren van hoogwaardige semantische kenmerken uit medische beelden. Door gebruik te maken van deze capaciteit, benut het voorgestelde uitlegbare deep learning-framework voor de classificatie van huidlaesies (EDLF-SLC) meerdere vooraf getrainde CNN-modellen om superieure prestaties te behalen. Om de noodzakelijke transparantie van het medische besluitvormingsproces te waarborgen, is LIME in de voorgestelde benadering geïmplementeerd om lokale, voor mensen leesbare verklaringen voor de resultaten te genereren. Het volledige framework kan worden onderverdeeld in drie hoofdfasen, zoals geïllustreerd in Figuur 2, namelijk: (1) datapreprocessing, (2) kenmerkextractie en classificatie, en (3) interpreteerbaarheid.
Modelarchitectuur en integratie
Als voorbereidende stap werden zeven populaire deep learning-modellen (MobileNetV2, ResNet50, EfficientNetB0, Xception, InceptionResNetV2, NASNetLarge en MobileNet) geselecteerd en afzonderlijk geëvalueerd op de validatiedataset, waarbij de vier meest effectieve modellen (ResNet50, EfficientNetB0, MobileNet en Xception) zijn gekozen voor de fase van kenmerkextractie. In het voorgestelde raamwerk is elke inputafbeelding x onafhankelijk door de geselecteerde pretrained modellen geleid. De laatste volledig verbonden laag is uit elk van deze modellen verwijderd, waardoor kenmerkvectoren uit de voorgaande lagen zijn verkregen. fResNet50(x), fEfficientNetB0(x), fMobileNet(x) en fXception(x) verwijzen naar de output-kenmerkvectoren van elk van de bovengenoemde modellen. Door deze kenmerkvectoren samen te voegen, wordt een nieuwe geaggregeerde kenmerkvector F(xi) verkregen:
F(xi) = [f(ResNet50)(xi);f(EfficientNetB0)(xi);f(MobileNet)(xi);f(Xception)(xi)] (1)
Vervolgens is F(xi) door een SVM-classifier met een radial basis function (RBF)-kernel geleid om het classificatieresultaat te verkrijgen. Het volledige algoritme voor het voorgestelde framework staat beschreven in Aanvullend Bestand 1.
Het voorgestelde raamwerk maakt gebruik van directe fusie op kenmerkniveau, omdat elke vooraf getrainde CNN-architectuur verschillende onderscheidende kenmerken van huidlaesies vastlegt via zijn specifieke netwerkarchitectuur en aangeleerde kenmerkhierarchie. Bijgevolg behoudt het samenvoegen van deze heterogene kenmerkrepresentaties complementaire informatie, wat bijdraagt aan een meer uitgebreide karakterisering van de laesies voorafgaand aan de classificatie. Hoewel dimensiereductietechnieken zoals principale componentenanalyse (PCA) of kenmerkselectie op basis van wederzijdse informatie de dimensionaliteit van de gefuseerde representatie kunnen verminderen, is de volledige gefuseerde kenmerkvector bewust behouden; de dimensionaliteit hiervan blijft namelijk computationeel beheersbaar voor de gebruikte RBF-SVM-classifier, terwijl de complementaire diagnostische informatie die door de verschillende CNN-backbones is geëxtraheerd, behouden blijft. Dit ontwerp geeft daarom prioriteit aan het behouden van complementaire diepe representaties in plaats van het elimineren van potentieel informatieve kenmerken vóór de classificatie.
Datapreparetie
De datapreparatie omvatte preprocessing en het splitsen van de dataset in trainings- en testsets. Preprocessing is gericht op het verbeteren van de datakwaliteit door inconsistenties te elimineren, dubbele elementen te verwijderen, inputafbeeldingen te formatteren en feature scaling uit te voeren voor een stabiele training van een goed model. Alle afbeeldingen zijn genormaliseerd naar het bereik van [−1, 1] middels Z-score normalisatie:
(2)
waarbij µ en σ respectievelijk staan voor de gemiddelde waarde en de standaarddeviatie van de overeenkomstige afbeelding. De dataset is verdeeld in twee subsets, namelijk een trainingsset (70,0%) en een testset (30,0%).
Data-augmentatie
Om overfitting te voorkomen en het generalisatievermogen van het model te vergroten, zijn enkele augmentaties toegepast op de trainingsset. Beeldaugmentaties houden in dat afbeeldingen worden gewijzigd om de dataset kunstmatig uit te breiden zonder de essentiële kenmerken van medische aandoeningen te veranderen. De augmentatie-pipeline is opgebouwd met behulp van de Keras Sequential API. Er is gebruikgemaakt van transformaties zoals willekeurige horizontale spiegeling, rotatie binnen een kleine hoek, herschalen, schalen, aanpassing van helderheid/contrast, zoomen en enkele kleine verschuivingen. Representatieve voorbeelden van geaugmenteerde afbeeldingen zijn weergegeven in Figuur 3.
Pretrained modellen
Verschillende pretrained deep learning-modellen zijn toegepast in het voorgestelde raamwerk om kenmerken uit de invoerbeelden te extraheren. Deze modellen omvatten MobileNet, ResNet50, EfficientNetB0, Xception, NASNetLarge, Inception-ResNet-v2 en MobileNetV2. MobileNet en MobileNetV2 zijn lichtgewicht deep learning-modellen voor efficiënte berekeningen met behulp van depth-wise separable convolutions. ResNet50 maakt gebruik van het mechanisme van residual connections voor het trainen van het model, waardoor het probleem van vanishing gradients tijdens de training wordt voorkomen. EfficientNetB0 slaat een brug tussen efficiëntie en precisie via de aanpak van compound scaling. Xception breidt het Inception-netwerk uit met behulp van depthwise separable convolutions. NASNetLarge maakt gebruik van neural architecture search om optimale deep neural network-structuren te bouwen, en Inception-ResNet-v2 vormt een hybride van het Inception-model en het mechanisme van residual connections. Kenmerkvectoren geëxtraheerd uit ResNet50 (1.024 kenmerken), EfficientNetB0 (1.280 kenmerken), MobileNet (1.024 kenmerken) en Xception (2.048 kenmerken) worden geconcateneerd om een uniforme 5.376-dimensionale representatie te produceren. Deze fusiestrategie is gekozen om complementaire representaties die door de verschillende CNN-architecturen zijn geleerd te behouden, in plaats van de representatie vóór de classificatie te reduceren. De resulterende kenmerkvector wordt vervolgens aangeboden aan de RBF-SVM-classifier, die de optimale niet-lineaire beslissingsgrens binnen de gefuseerde kenmerkruimte bepaalt.
Uitlegbaar AI-model (LIME)
Om enige lokale interpreteerbaarheid van de modelvoorspellingen te bieden, maken de auteurs gebruik van de methode waarbij lokale, voor mensen leesbare verklaringen worden gegenereerd voor elke specifieke modelvoorspelling, genaamd LIME47. Voor elke invoerafbeelding verdeelt LIME deze eerst in kleinere eenheden, genaamd superpixels. Vervolgens worden er perturbaties gegenereerd op basis van de oorspronkelijke afbeelding, en worden de voorspellingen van het deep neural model geschat voor elke perturbatie. Daarna wordt een gewogen lineair model gefit op de perturbaties, waarbij gebruik wordt gemaakt van gewichten die afhangen van hun nabijheid tot de oorspronkelijke invoerinstantie. Na het fitten van het lineaire model worden de scores voor feature-belangrijkheid geschat, die de rol van elke superpixel in de voorspelling van het uiteindelijke label aangeven. Deze belangrijkheidsscores worden visueel geaccentueerd in de uiteindelijke verklarende afbeelding. Het LIME-algoritme is weergegeven in Aanvullend Bestand 2.