WFE-YOLO: A LIGHTWEIGHT PIG BEHAVIOR DETECTION MODEL FOR LIVESTOCK FARMING APPLICATIONS
WFE-YOLO:面向养殖场景的轻量化猪只行为检测模型
DOI : https://doi.org/10.35633/inmateh-78-99
Authors
Abstract
In a real-life breeding environment, fine-grained pig behavior detection is of great significance for health assessment, welfare monitoring, and intelligent management. However, issues such as dense target distribution, severe occlusion, subtle inter-class differences, class imbalance, and limited computing resources make it difficult to achieve both detection accuracy and computational efficiency. To address these problems, this paper proposes a lightweight pig behavior detection model WFE-YOLO based on YOLOv11, which is used to identify five typical behaviors: standing, lying down, eating, drinking, and chewing. This method conducts collaborative optimization at three levels: training sample distribution, feature representation, and detection head design. Specifically, a weighted sampling strategy is adopted to enhance the learning sufficiency of low-frequency behaviors; a lightweight gated feature extraction module is introduced to improve the fine-grained representation ability; an efficient detection head is designed to reduce structural redundancy and computational overhead. Experimental results show that on the data set of this paper, the Precision, Recall, and mAP@50 of WFE-YOLO reach 0.8154, 0.7803, and 0.8233 respectively; compared with YOLOv11n, the parameter size is reduced from 2.58M to 1.96M, GFLOPs is reduced from 6.3G to 4.4G, and FPS reaches 520.43. Under the experimental settings adopted in this paper, compared with several mainstream YOLO models, WFE-YOLO demonstrates a better balance between detection performance and model complexity, especially in low-frequency and easily confused behaviors such as Drink and Bite, and has a greater advantage. These results indicate that WFE-YOLO provides a lightweight, application-oriented solution for pig behavior monitoring in complex breeding environments, with strong potential for deployment on edge devices.
Abstract in English



