Evaluation of two CNN models, VGGNet-16 & VGGNet-19, for classification of Alzheimer’s disease in brain MRI scans

• Main Authors: Jonsson, Tim, Tapper, Isabella
• Format: Others
• Language: English
• Published: KTH, Skolan för elektroteknik och datavetenskap (EECS) 2020
• Subjects: Computer Sciences; Datavetenskap (datalogi)
• Online Access: http://urn.kb.se/resolve?urn=urn:nbn:se:kth:diva-280141

Computer-aided-diagnosis (CAD) emerged in the early 1950s and since then CAD has facilitated the diagnosing of many medical conditions and diseases. In particular, CADfor Alzheimer’s disease (AD) has been immensely researched the last decade thanks to advanced neuroimaging techniques such as magnetic resonance imaging (MRI) and positron emission tomography (PET). Today around 44 million people worldwide have AD and researchers hope to discover accurate ways to detect AD before the symptoms begin. There are currently no validated so-called biological markers (biomarkers) for AD, meaning that there are no reliable indicators that can accurately diagnose AD. However, according to experts, machine learning and neuroimaging is among the most promising areas of research focused on biomarkers and early diagnosis of AD. The state-of-the-art machine learning method for image classification are convolutional neural networks (CNNs). At a recent study at Bharati Vidyapeeth’s College of Engineering and Karunya University, a convolutional neural network VGGNet-16 was used in an experiment in order to correctly classify AD using MRI scans. Experimentation was performed on data collected from the Alzheimer’s Disease Neuroimaging Initiative (ADNI) database. The accuracy using the described method was 95.73% for the validation set. The purpose of this bachelor thesis was to compare two different convolution neural network models: VGGNet-16 and VGGNet-19, comparing their results and performances for classifying AD using MRI scans from ADNI database. Sets of images were elected, some include and some exclude the hippocampus, since AD starts spreading in the hippocampus. Using transfer learning, the CNN models were trained with (a) random validation split, (b) cross validation and (c) different slice range not including the hippocampus. The results of this study show that the models were good at classifying true-negative, which is diagnosing a healthy patient as healthy. Hippocampus seems to be a promising biomarker for AD because experiment (c) achieved a lower accuracy than (a) and (b). In conclusion there is no real statistically proven difference between VGGNet-16 and VGGNet-19. Even then, this thesis showed that simpler CNN architectures can be utilized to classify AD with equally mild success rate on a very limited dataset. The two CNN models’ accuracy were between 66.6- 74.8% for classifying AD depending on the training approach. === Datorstödd diagnostisk (CAD) uppkom under tidigt 50-tal och har sedan dess använts för att diagnostisera många medicinska tillstånd och sjukdomar. Specifikt CAD för Alzheimers sjukdom (AD) har undersökts kraftigt det senaste decenniet till följd av uppkomsten av avancerade hjärnavbildningstekniker såsom Magnetic Resonanse Imaging (MRI) och Positron Emission Tomography (PET). I dagsläget lider 44 miljoner människor av AD. Forskare hoppas i framtiden kunna upptäcka sjukdomen i ett tidigt stadie, men i dagsläget finns ingen pålitlig indikator som med god säkerhet kan klassificera AD. Enligt experter är dock maskininlärning och hjärnavbildningstekniker de mest lovande områdena för tidig diagnostik av AD. Den masknikinlärningsmodell som ligger i framkant för bildigenkänning är faltningsnätverk (CNN). Vid en ny studie av Bharati Vidyapeeth’s College of Engineering och Karunya University användes ett CNN, VGG-16, för att klassificera AD med hjälp av MRI-bilder. Experimentet utfördes på data från Alzheimer’s Disease Neuroimaging Initiative (ADNI) och uppnådde en träffsäkerhet på 95.73%. Syftet med vår studie var att utvärdera två CNN-modeller, VGGNet-16 och VGGNet-19, för att jämföra deras resultat och prestanda vid klassificering av AD med bilder från ADNI-databasen. Uppsättningar av bilder valdes varav hippocampus inkluderades i vissa och exkluderades i andra, detta då AD tros börja i hippocampus. Med överförningsinlärning tränades CNN modellerna på (a) slumpmässigt utvalt valideringsdata, (b) korsvalidering, och (c) bilder utan hippocampus. Resultatet visade att modellerna var bra på att klassificera sanna-negativa, d.v.s. friska patienter klassas som friska. Därefter visade även resultatet att modellerna uppnådde en högre träffsäkerhet i experiment (a) och (b) än i (c). Detta medför att hippocampus kan ses som en användbar biomarkör. Slutligen visade resultatet att modellerna statistiskt sett inte kan urskiljas från varandra, vilket kan tyda på att de presterar lika. Dock visade denna studie att simpla CNN-modeller kan användas för att klassificera AD på väldigt begränsad mängd data. De två modellerna uppnådde en träffsäkerhet på mellan 66,6% – 74,8% vid klassificering av AD beroende på hur modellerna tränats.