RCS-YOLO: A Fast and High-Accuracy Object Detector for Brain Tumor Detection

Ming Kang,Chee-Ming Ting,Fung Fung Ting,Raphaël C.-W. Phan
DOI: https://doi.org/10.1007/978-3-031-43901-8_57
2023-10-03
Abstract:With an excellent balance between speed and accuracy, cutting-edge YOLO frameworks have become one of the most efficient algorithms for object detection. However, the performance of using YOLO networks is scarcely investigated in brain tumor detection. We propose a novel YOLO architecture with Reparameterized Convolution based on channel Shuffle (RCS-YOLO). We present RCS and a One-Shot Aggregation of RCS (RCS-OSA), which link feature cascade and computation efficiency to extract richer information and reduce time consumption. Experimental results on the brain tumor dataset Br35H show that the proposed model surpasses YOLOv6, YOLOv7, and YOLOv8 in speed and accuracy. Notably, compared with YOLOv7, the precision of RCS-YOLO improves by 1%, and the inference speed by 60% at 114.8 images detected per second (FPS). Our proposed RCS-YOLO achieves state-of-the-art performance on the brain tumor detection task. The code is available at <a class="link-external link-https" href="https://github.com/mkang315/RCS-YOLO" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition,Signal Processing,Applications,Machine Learning
What problem does this paper attempt to address?
This paper aims to address the problem of fast and high-precision object detection in brain tumor detection. Specifically, the paper proposes a novel YOLO architecture—RCS-YOLO, which combines Reparameterized Convolution (RCS) based on Channel Shuffle, and demonstrates its superior performance in brain tumor detection tasks through experiments. ### Main Objectives: 1. **Improve Detection Speed and Accuracy**: By introducing new modules RCS (Reparameterized Convolution with Channel Shuffle) and RCS-One Shot Aggregation (RCS-OSA), achieve faster inference speed and higher detection accuracy. 2. **Optimize Computational Efficiency**: By reducing the number of convolutional layers and computational complexity, lower memory consumption, thereby enhancing overall computational efficiency. 3. **Application in Medical Image Detection**: Apply the improved YOLO model to brain tumor detection and verify its effectiveness and practicality in real-world applications. ### Experimental Results: The paper tested on the public brain tumor dataset Br35H, and the results show that RCS-YOLO outperforms existing models such as YOLOv6, YOLOv7, and YOLOv8 in terms of detection accuracy and speed. Specifically: - Detection accuracy increased by 1%, inference speed improved by 60%, reaching 114.8 images per second (FPS). ### Summary: This paper proposes a new YOLO variant—RCS-YOLO, which significantly improves the model's inference speed while maintaining high accuracy by introducing RCS and RCS-OSA modules, making it suitable for fast and accurate detection of brain tumors.