Automated multi-class wound assessment using dedicated instance segmentation models for boundary detection and classification

arXiv:2603.27325v2 Announce Type: replace-cross Abstract: Accurate wound classification (WC) and boundary segmentation are essential for guiding clinical decisions in chronic and acute wound management. However, most existing artificial intelligence (AI) models are limited, focusing on a narrow set of wound types, limited variations in wound severity, or a single task (segmentation or classification), which reduces their clinical applicability. This study presents two dedicated instance segmentation models based on You Only Look Once (YOLO)v11 that perform wound boundary segmentation (WBS) and WC across five clinically relevant wound types: burn injury (BI), pressure injury, diabetic foot ulcer, vascular ulcer, and surgical wound. A wound-type balanced dataset of 2,963 annotated images was created to train the models for both tasks, using five-fold cross-validation. Models trained on the original, non-augmented dataset performed consistently across folds, though BI detection accuracy was relatively low; augmenting the dataset with rotation, flipping, and variations in brightness, saturation, and exposure significantly improved performance, particularly for visually subtle BI cases. Among the tested variants, YOLOv11x achieved the best WBS performance (F1-score: 0.9341; mAP50: 0.9629). For WC, YOLOv11m achieved the highest mAP50 (0.9194) and mAP50-95 (0.6950), whereas YOLOv11l achieved the highest F1-score (0.8797). The lightweight YOLOv11n provided comparable accuracy at lower computational cost, making it suitable for resource-constrained deployments. Supported by confusion matrices and visual detection outputs, the results confirm robustness against complex backgrounds and high intra-class variability, demonstrating the potential of YOLOv11-based architectures for accurate, real-time wound analysis in clinical and remote care settings.

Abstract:Accurate wound classification (WC) and boundary segmentation are essential for guiding clinical decisions in chronic and acute wound management. However, most existing artificial intelligence (AI) models are limited, focusing on a narrow set of wound types, limited variations in wound severity, or a single task (segmentation or classification), which reduces their clinical applicability. This study presents two dedicated instance segmentation models based on You Only Look Once (YOLO)v11 that perform wound boundary segmentation (WBS) and WC across five clinically relevant wound types: burn injury (BI), pressure injury, diabetic foot ulcer, vascular ulcer, and surgical wound. A wound-type balanced dataset of 2,963 annotated images was created to train the models for both tasks, using five-fold cross-validation. Models trained on the original, non-augmented dataset performed consistently across folds, though BI detection accuracy was relatively low; augmenting the dataset with rotation, flipping, and variations in brightness, saturation, and exposure significantly improved performance, particularly for visually subtle BI cases. Among the tested variants, YOLOv11x achieved the best WBS performance (F1-score: 0.9341; mAP50: 0.9629). For WC, YOLOv11m achieved the highest mAP50 (0.9194) and mAP50-95 (0.6950), whereas YOLOv11l achieved the highest F1-score (0.8797). The lightweight YOLOv11n provided comparable accuracy at lower computational cost, making it suitable for resource-constrained deployments. Supported by confusion matrices and visual detection outputs, the results confirm robustness against complex backgrounds and high intra-class variability, demonstrating the potential of YOLOv11-based architectures for accurate, real-time wound analysis in clinical and remote care settings.

Author's version of the peer-reviewed article published open access (CC BY 4.0) in Artificial Intelligence in Health, online 7 September 2026. 30 pages, 8 figures, 6 tables. v2: title, abstract and text updated to match the published version (v1 title: "Improving Automated Wound Assessment Using Joint Boundary Segmentation and Multi-Class Classification Models")

Computer Vision and Pattern Recognition (cs.CV); Artificial Intelligence (cs.AI)

arXiv:2603.27325 [cs.CV]

(or arXiv:2603.27325v2 [cs.CV] for this version)

https://doi.org/10.36922/AIH026250065 DOI(s) linking to related resources

analiza aiqbench

Automated multi-class wound assessment using dedicated instance segmentation models for boundary detection and classification

Subiecte conexe