ℓ1DecNet+ : A new architecture framework by ℓ1 decomposition and iteration unfolding for sparse feature segmentation
Research output: Working Papers › Preprint
Author(s)
Related Research Unit(s)
Detail(s)
Original language | English |
---|---|
Number of pages | 12 |
Publication status | Online published - 16 Jun 2024 |
Link(s)
DOI | DOI |
---|---|
Permanent Link | https://scholars.cityu.edu.hk/en/publications/publication(453ed45e-1202-4451-af6c-c8aec500eb32).html |
Abstract
ℓ1 based sparse regularization plays a central role in compressive sensing and image processing. In this paper, we propose ℓ1DecNet, as an unfolded network derived from a variational decomposition model incorporating ℓ1 related sparse regularization and solved by scaled alternating direction method of multipliers (ADMM). ℓ1DecNet effectively decomposes an input image into a sparse feature and a learned dense feature, and thus helps the subsequent sparse feature related operations. Based on this, we develop ℓ1DecNet+, a learnable architecture framework consisting of our ℓ1DecNet and a segmentation module which operates over extracted sparse features instead of original images. This architecture combines well the benefits of mathematical modeling and data-driven approaches. To our best knowledge, this is the first study to incorporate mathematical image prior into feature extraction in segmentation network structures. Moreover, our ℓ1DecNet+ framework can be easily extended to 3D case. We evaluate the effectiveness of ℓ1DecNet+ on two commonly encountered sparse segmentation tasks: retinal vessel segmentation in medical image processing and pavement crack detection in industrial abnormality identification. Experimental results on different datasets demonstrate that, our ℓ1DecNet+ architecture with various lightweight segmentation modules can achieve equal or better performance than their enlarged versions respectively. This leads to especially practical advantages on resource-limited devices.
Research Area(s)
- eess.IV, cs.CV, math.OC
Bibliographic Note
12 pages, 7 figures
Citation Format(s)
ℓ1DecNet+: A new architecture framework by ℓ1 decomposition and iteration unfolding for sparse feature segmentation. / Ren, Yumeng; Gao, Yiming; Tai, Xue-cheng et al.
2024.
2024.
Research output: Working Papers › Preprint