Classification of acute myeloid leukemia by pre-trained deep neural networks: A comparison with different activation functions.
Journal:
Medical engineering & physics
PMID:
39922655
Abstract
Acute Myeloid Leukemia(AML) is a rapidly progressing cancer affecting blood and bone marrow, marked by the swift proliferation of abnormal myeloid cells. Effective treatment requires precise classification of AML subtypes. Conventional classification methods rely on manual microscopic analysis, which is time-consuming and variable, while traditional machine learning approaches often struggle with feature extraction and generalization. This underscores the need for automated detection methods which enhance this process by minimizing manual effort. Activation functions are critical in neural networks, introducing non-linearity that influences training convergence, computational efficiency, and model performance. This study evaluates the effectiveness of CNN architectures, Sequential (VGG16), Directed Acyclic Graph (InceptionV3), and Residual (ResNet50v2) in distinguishing between AML subtypes: AML without maturation, Acute Monoblastic Leukemia, and Pure Erythroid Leukemia, using peripheral blood smear images, while also investigating the impact of different activation functions on model accuracy and training time. The results show that ResNet50v2 achieves the shortest training time, while InceptionV3 takes the longest due to its complex architecture. GELU delivers the highest accuracy, reaching 94.02 % in InceptionV3 and 92.53 % in ResNet50v2, while SELU achieves the highest accuracy for VGG16 at 92.83 %. Mish provides competitive accuracy with lower training time than GELU, while Softplus and Softsign consistently perform poorly. This research demonstrates the potential of CNNs for automating AML subtype classification and identifies GELU as the most effective activation function. Future work could explore data augmentation, optimized activation functions, and attention mechanisms to improve classification performance.