Skip to main content

METHODS article

Front. Artif. Intell.
Sec. Medicine and Public Health
Volume 7 - 2024 | doi: 10.3389/frai.2024.1455331

Enhancing Random Forest Predictive Performance for Foot and Mouth Disease Outbreaks in Uganda: A Calibrated Uncertainty Prediction Approach for Varying Distributions

Provisionally accepted
Geofrey Kapalaga Geofrey Kapalaga 1*Florence N. Kivunike Florence N. Kivunike 1Susan D. Kerfua Susan D. Kerfua 2Daudi Jjingo Daudi Jjingo 1Savino Biryomumaisho Savino Biryomumaisho 3Justus Rutaisire Justus Rutaisire 2PAUL SSAJJAKAMBWE PAUL SSAJJAKAMBWE 2Swidiq Mugerwa Swidiq Mugerwa 2Seguya Abbey Seguya Abbey 2Mulindwa H. Aaron Mulindwa H. Aaron 2Yusuf Kiwala Yusuf Kiwala 4
  • 1 College of Computing and Information Science, Makerere University, Kampala, Uganda
  • 2 National Livestock Resources Research Institute, Tororo, Uganda
  • 3 College of Veterinary Medicine, Animal Resources and Biosecurity, Makerere University, Kampala, Uganda
  • 4 College of Business and Management Sciences (CoBAMS), Makerere University, Kampala, Uganda

The final, formatted version of the article will be published soon.

    Foot-and-mouth disease poses a significant threat to both domestic and wild cloven-hoofed animals, leading to severe economic losses and jeopardizing food security. While machine learning models have become essential for predicting foot-and-mouth disease outbreaks, their effectiveness is often compromised by distribution shifts between training and target datasets, especially in non-stationary environments. Despite the critical impact of these shifts, their implications in foot-and-mouth disease outbreak prediction have been largely overlooked. This study introduces the Calibrated Uncertainty Prediction approach, designed to enhance the performance of Random Forest models in predicting foot-and-mouth disease outbreaks across varying distributions. The Calibrated Uncertainty Prediction approach effectively addresses distribution shifts by calibrating uncertain instances for pseudo-label annotation, allowing the active learner to generalize more effectively to the target domain. By utilizing a probabilistic calibration model, Calibrated Uncertainty Prediction pseudo-annotates the most informative instances, refining the active learner iteratively and minimizing the need for human annotation and outperforming existing methods known to mitigate distribution shifts. This reduces costs, saves time, and lessens the dependence on domain experts while achieving outstanding predictive performance. The results demonstrate that Calibrated Uncertainty Prediction significantly enhances predictive performance in non-stationary environments, achieving an accuracy of 98.5%, Area Under the Curve of 0.842, recall of 0.743, precision of 0.855, and an F1 score of 0.791. These findings underscore Calibrated Uncertainty Prediction's ability to overcome the vulnerabilities of existing ML models, offering a robust solution for foot-and-mouth disease outbreak prediction and contributing to the broader field of predictive modeling in infectious disease management.

    Keywords: Foot-and-Mouth Disease, random forest, Distribution shifts, performance improvement rates, calibrated uncertainty prediction

    Received: 10 Jul 2024; Accepted: 30 Aug 2024.

    Copyright: © 2024 Kapalaga, Kivunike, Kerfua, Jjingo, Biryomumaisho, Rutaisire, SSAJJAKAMBWE, Mugerwa, Abbey, Aaron and Kiwala. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence: Geofrey Kapalaga, College of Computing and Information Science, Makerere University, Kampala, Uganda

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.