Open AccessTransfer learning for cancer diagnosis in histopathological images
Author(s) -
Sandhya Aneja,
Nagender Aneja,
Pg Emeroylariffion Abas,
Abdul Ghani Naim
Publication year - 2022
Publication title -
iaes international journal of artificial intelligence
Language(s) - English
Resource type - Journals
eISSN - 2252-8938
pISSN - 2089-4872
DOI - 10.11591/ijai.v11.i1.pp129-136
Subject(s) - computer science , transfer of learning , task (project management) , artificial intelligence , recall , extractor , machine learning , feature (linguistics) , exploit , sensitivity (control systems) , precision and recall , transfer (computing) , pattern recognition (psychology) , linguistics , philosophy , computer security , management , electronic engineering , process engineering , engineering , economics , parallel computing
Transfer learning allows us to exploit knowledge gained from one task to assist in solving another but relevant task. In modern computer vision research, the question is which architecture performs better for a given dataset. In this paper, we compare the performance of 14 pre-trained ImageNet models on the histopathologic cancer detection dataset, where each model has been configured as naive model, feature extractor model, or fine-tuned model. Densenet161 has been shown to have high precision whilst Resnet101 has a high recall. A high precision model is suitable to be used when follow-up examination cost is high, whilst low precision but a high recall/sensitivity model can be used when the cost of follow-up examination is low. Results also show that transfer learning helps to converge a model faster.