An Adaptive Early Stopping Technique for DenseNet169-Based Knee Osteoarthritis Detection Model

Bander Al-rimy* (Corresponding / Lead Author), Faisal Saeed (Corresponding / Lead Author), Mohammed Al-Sarem (Corresponding / Lead Author), Abdullah Albarrak (Corresponding / Lead Author), Sultan Qasem (Corresponding / Lead Author)

*Corresponding author for this work

    Research output: Contribution to journalArticlepeer-review

    6 Citations (SciVal)

    Abstract

    Knee osteoarthritis (OA) detection is an important area of research in health informatics that aims to improve the accuracy of diagnosing this debilitating condition. In this paper, we investigate the ability of DenseNet169, a deep convolutional neural network architecture, for knee osteoarthritis detection using X-ray images. We focus on the use of the DenseNet169 architecture and propose an adaptive early stopping technique that utilizes gradual cross-entropy loss estimation. The proposed approach allows for the efficient selection of the optimal number of training epochs, thus preventing overfitting. To achieve the goal of this study, the adaptive early stopping mechanism that observes the validation accuracy as a threshold was designed. Then, the gradual cross-entropy (GCE) loss estimation technique was developed and integrated to the epoch training mechanism. Both adaptive early stopping and GCE were incorporated into the DenseNet169 for the OA detection model. The performance of the model was measured using several metrics including accuracy, precision, and recall. The obtained results were compared with those obtained from the existing works. The comparison shows that the proposed model outperformed the existing solutions in terms of accuracy, precision, recall, and loss performance, which indicates that the adaptive early stopping coupled with GCE improved the ability of DenseNet169 to accurately detect knee OA.
    Original languageEnglish
    JournalDiagnostics
    Publication statusPublished (VoR) - 29 May 2023

    Fingerprint

    Dive into the research topics of 'An Adaptive Early Stopping Technique for DenseNet169-Based Knee Osteoarthritis Detection Model'. Together they form a unique fingerprint.

    Cite this