Ensemble Learning Applied to Classification of Malignant and Benign Breast Cancer
2021 (English)Independent thesis Basic level (degree of Bachelor), 10 credits / 15 HE credits
Student thesisAlternative title
Samlingsinlärning applicerat på klassifikation av elakartad och godartad bröstcancer (Swedish)
Abstract [en]
In this study, we show how ensemble learning can be useful for the future of breast cancer diagnosis. The chosen ensemble learning method was bagging, which made use of the classifiers Support Vector Machine (SVM), Decision Tree (DT) and Naive Bayes (NB) in order to classify mammograms as benign or malignant. The results achieved with bagging were compared to the results of each individual classifier previously mentioned. Overall, the results showed that the benefits of ensemble learning were varying, dependent on certain factors. Affecting aspects were: which classifier that was used, chosen method for extracting input data, but also which tumor types that were used in training and evaluation of each classifier. While classification using DT improved significantly with bagging, SVM and NB gave negligible performance benefits. Finally, this study only scratched the surface of known ensemble learning methods, indicating that there may be a lot of room for future research in the area.
Abstract [sv]
I denna rapport visar vi hur samlingsinlärning kan vara användbart för framtida diagnostisering av bröstcancer. Den valda samlingsinlärning-metoden var bagging", vilket tog användning av Support Vector Machine (SVM), Decision Tree (DT) och Naive Bayes (NB) för att klassificera mammogram som godartade eller elakartade. Resultaten som togs fram för bagging"jämfördes avslutligen med resultaten från respektive ovannämnd klassifierare. Generellt visade resultaten att fördelarna med samlingsinlärning var varierande, beroende på vissa faktorer. Påverkande aspekter var: vilken klassifierare som användes, vald metod för extraktion av inmatningsdata, men också vilka tumörtyper som användes för träning och evaluering av respektive klassifierare. Medans klassifikation med DT förbättrades signifikant med bagging", var skillnaderna försumbara med SVM och NB. Slutligen, skrapar denna studie enbart på ytan av kända samlingsinlärning-metoder, vilket indikerar att det kan finnas mycket utrymme för framtida forskning i området.
Place, publisher, year, edition, pages
2021. , p. 51
Series
TRITA-EECS-EX ; 2021:495
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-302551OAI: oai:DiVA.org:kth-302551DiVA, id: diva2:1597706
Subject / course
Computer Science
Educational program
Master of Science in Engineering - Computer Science and Technology
Supervisors
Examiners
2021-09-282021-09-272022-06-25Bibliographically approved