mlsanityData to 14 Oct 2026

WILLIE: A Unified Framework and Benchmark for Wound Classification, Segmentation, and Localization

arXiv 2610.05341 ↗First posted 4 Oct 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

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% [email protected] 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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 46 references that are in the map. First posted on arXiv 4 Oct 2026.

Builds on

19 of 46 references.

References · 19

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  2. 2023Segment Anythingcited by 8,234
  3. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  4. 2016Feature Pyramid Networks for Object Detectioncited by 5,036
  5. 2022A ConvNet for the 2020scited by 3,579
  6. 2024SAM 2: Segment Anything in Images and Videoscited by 2,863
  7. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
  8. 2019FCOS: Fully Convolutional One-Stage Object Detectioncited by 1,577
  9. 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,358
  10. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  11. 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
  12. 2023DETRs Beat YOLOs on Real-time Object Detectioncited by 533
7 more
  1. 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 375
  2. 2021Efficiently Identifying Task Groupings for Multi-Task Learningcited by 222
  3. 2022FUSeg: The Foot Ulcer Segmentation Challengecited by 22
  4. 2018Deep Learning with Mixed Supervision for Brain Tumor Segmentationcited by 18
  5. 2023FUSegNet: A Deep Convolutional Neural Network for Foot Ulcer Segmentationcited by 13
  6. 2020Image Based Artificial Intelligence in Wound Assessment: A Systematic Reviewcited by 11
  7. 2023Integrated Image and Location Analysis for Wound Classification: A Deep Learning Approachcited by 9

Built on by

Not cited yet.