Development of a multi-task learning framework with gradnorm for precise wound tissue analysis

Hyunyoung Kang, Byungho Oh, Solam Lee, YuSung Chu, Jiye Kim and Sejung Yang

PLOS ONE, 2026, vol. 21, issue 2, 1-20

Abstract: Chronic wounds impose a substantial burden on patients and healthcare systems, necessitating accurate qualification of precise wound analysis for effective diagnosis and treatment. Wound size and the proportional composition of internal tissues are critical indicators of healing progression. Traditional segmentation approaches such as Separate Task Learning (STL) suffer from parameter inefficiency, while Multi-Task Learning (MTL), though efficient, often experiences task imbalance that leads to performance degradation in specific tasks. To overcome these challenges, this study proposes WING-MTL (Wound and Wound Tissue Integrated with Gradient Normalization Multi-Task Learning), a novel MTL framework that dynamically balances gradient magnitudes across tasks to enhance accuracy and training stability. Implemented on an Attention-UNet backbone,and incorporates Gradient Normalization to adjust learning gradients in real time, ensuring balanced optimization for wound and wound tissue segmentation tasks. Quantitative evaluations demonstrate that WING-MTL yields statistically significant improvements over STL and outperforms conventional and advanced MTL methods. Analysis of validation loss revealed convergence of both tasks at the same epoch, indicating balanced learning. Furthermore, improved segmentation performance was observed compared to both STL and MTL approaches, particularly for challenging wound tissue types such as slough and epithelium in qualitative analysis. These findings confirm that WING-MTL addresses the task imbalance inherent in MTL frameworks while maintaining parameter efficiency. WING-MTL was evaluated across diverse backbones, including not only UNet-based model but also Resnet and Transformer-based architectures to validate the architectural flexibility of the proposed framework. Consistent performance improvement across these varied architectures in wound tissue segmentation demonstrates the broad applicability. Furthermore, longitudinal analysis of chronic wound patients was conducted to assess the clinical utility of WING-MTL in real-world scenarios. The framework presents a promising and accurate approach for tracking wound healing progression and serves as a potential adjunct for clinical decision-making in chronic wound care.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0340258 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 40258&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0340258

DOI: 10.1371/journal.pone.0340258

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().