Enhancing the YOLOv8 model for realtime object detection to ensure online platform safety.

Journal: Scientific reports
Published Date:

Abstract

In today's digital environment, effectively detecting and censoring harmful and offensive objects such as weapons, addictive substances, and violent content on online platforms is increasingly important for user safety. This study introduces an Enhanced Object Detection (EOD) model that builds upon the YOLOv8-m architecture to improve the identification of such harmful objects in complex scenarios. Our key contributions include enhancing the cross-stage partial fusion blocks and incorporating three additional convolutional blocks into the model head, leading to better feature extraction and detection capabilities. Utilizing a public dataset covering six categories of harmful objects, our EOD model achieves superior performance with precision, recall, and mAP50 scores of 0.88, 0.89, and 0.92 on standard test data, and 0.84, 0.74, and 0.82 on challenging test cases-surpassing existing deep learning approaches. Furthermore, we employ explainable AI techniques to validate the model's confidence and decision-making process. These advancements not only enhance detection accuracy but also set a new benchmark for harmful object detection, significantly contributing to the safety measures across various online platforms.

Authors

  • Mohammed Kawser Jahan
    Department of Computer Science, American International University-Bangladesh, Dhaka, 1229, Bangladesh.
  • Fokrul Islam Bhuiyan
    Department of Computer Science, American International University-Bangladesh, Dhaka, 1229, Bangladesh.
  • Al Amin
    Department of Computer Science, American International University-Bangladesh, Dhaka, 1229, Bangladesh.
  • M F Mridha
    Department of Computer Science and Engineering, American International University, Dhaka, Bangladesh.
  • Mejdl Safran
    Department of Computer Science, College of Computer and Information Sciences, King Saud University, P.O. Box 51178, Riyadh 11543, Saudi Arabia.
  • Sultan Alfarhood
    Department of Computer Science, College of Computer and Information Sciences, King Saud University, P.O. Box 51178, Riyadh 11543, Saudi Arabia.
  • Dunren Che
    School of Computing, Southern Illinois University, Carbondale, IL 62901, USA.