WILLIE: A Unified Framework and Benchmark for Wound Classification, Segmentation, and Localization
cs.CV
Submitted: 2026-10-04
Updated: 2026-10-04
Comments: 21 pages, 4 figures, 9 tables. Published in Proceedings of the 11th Machine Learning for Healthcare Conference (MLHC 2026), PMLR 340:1243-1263. Code: https://github.com/Qian-Group-HRI/Willie
Journal ref: Proceedings of the 11th Machine Learning for Healthcare Conference, PMLR 340:1243-1263, 2026
Code: https://github.com/Qian-Group-HRI/Willie
License: http://creativecommons.org/licenses/by/4.0/
The gist: Chronic wound management affects over 8.2 million patients in the United States and imposes substantial clinical and economic burden.
Terminology
Abstract
Chronic wound management affects over 8.2 million patients in the United States and imposes substantial clinical and economic burden. Clinical wound assessment commonly involves three coupled tasks: identifying wound type, delineating wound boundaries, and localizing the wound region for measurement and monitoring. Despite this clinical coupling, existing machine learning approaches typically address wound classification, segmentation, and localization using separate models. We present WILLIE, a unified framework and benchmark for wound classification, segmentation and localization that enables systematic evaluation of multi-task wound analysis under a common protocol. WILLIE harmonizes three public wound datasets into a shared benchmark and compares unified models across three scaling configurations against 10 single-task baselines. The best model achieves 91.88% classification accuracy, 91.41% Dice, and 96.23% AP@0.5 while producing all three outputs in a single forward pass. Beyond aggregate performance, our results show that segmentation-derived localization outperforms dedicated detection baselines in this benchmark, suggesting that box-based localization may be unnecessary for spatially coherent wound targets. Our findings highlight that effective multi-task learning in healthcare imaging depends not only on shared representations, but also on task formulation, compatibility, and benchmark design.
Sources
- WoundNet-Ensemble: A Novel IoMT System Integrating Self-Supervised Deep Learning and Multi-Model Fusion for Automated, High-Accuracy Wound Classification and Healing Progression Monitoring
- SAM 2: Segment Anything in Images and Videos
- FUSeg: The Foot Ulcer Segmentation Challenge
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models