NIH Chest X-ray Classifier

Eksempler fra den holdt-utenfor testen

Velg et eksempel for å se resultater fra den valgte modellen etter sluttkontrollen. Dette er en prosjektvisning med lagrede testbilder, ikke et verktøy for medisinsk vurdering.

Viktig. Bildene er fra NIH ChestX-ray14 og brukes her som dokumentasjon av studentprosjektet. Hvert bilde har en registrert testetikett. Det betyr ikke at modellen stiller en diagnose fra bildet, og visningen tar ikke imot opplastinger.

112 120bilder i hele datasettarbeidet

25 596bilder i separat test

0,800gjennomsnittlig AUROC

Originalt røntgenbilde fra testsettet — Originalt testbilde

NIH sin annoterte fasit for kardiomegali — Fasit fra NIH-annotasjon

Grad-CAM for kardiomegali — Grad-CAM for valgt klasse

Velg et bilde og start den lagrede modellkjøringen.

Registrert testetikett

Kardiomegali

Forstørret hjerte. Dette er etiketten som følger dette holdt-utenfor testbildet.

Modellscore for valgt klasse0,997
Skåren er modellens rå utgang fra 0 til 1. Den er ikke kalibrert som sykdomssannsynlighet.

Tre høyeste modellskårer

Kardiomegali 0,997
Pleural effusjon 0,713
Infiltrat 0,572

Slik bør dette leses

Fasitbildet viser NIH sin annoterte avgrensning. Heatmapet viser områdene som påvirket modellskåren for den valgte klassen mest. Det er ikke en segmentering av sykdom og erstatter ikke en radiologisk vurdering.

Hva som ble gjort

En ResNet-50 med 14 utganger ble videreført fra et lagret kontrollpunkt og evaluert på NIH sin offisielle testliste. 77 785 bilder ble brukt i trening, 8 739 i pasientadskilt validering og 25 596 ble holdt utenfor til slutt. Den valgte modellen oppnådde gjennomsnittlig AUROC 0,800 på sluttsettet. Eksemplene på denne siden og Grad-CAM-bildene ble laget på nytt med den samme modellen.

Les prosjektsammendraget Se kode og dokumentasjon på GitHub