mlsanityData to 7 Oct 2026

Multi-Task Learning with Additive U-Net for Image Denoising and Classification

arXiv 2602.12649 ↗First posted 13 Feb 2026cs.CV cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

We investigate additive skip fusion in U-Net architectures for image denoising and denoising-centric multi-task learning (MTL). By replacing concatenative skips with gated additive fusion, the proposed Additive U-Net (AddUNet) constrains shortcut capacity while preserving fixed feature dimensionality across depth. This structural regularization induces controlled encoder-decoder information flow and stabilizes joint optimization. Across single-task denoising and joint denoising-classification settings, AddUNet achieves competitive reconstruction performance with improved training stability. In MTL, learned skip weights exhibit systematic task-aware redistribution: shallow skips favor reconstruction, while deeper features support discrimination. Notably, reconstruction remains robust even under limited classification capacity, indicating implicit task decoupling through additive fusion. These findings show that simple constraints on skip connections act as an effective architectural regularizer for stable and scalable multi-task learning without increasing model complexity.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 12 references that are in the map. First posted on arXiv 13 Feb 2026.

Builds on

10 of 12 references.

References · 10

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2018CBAM: Convolutional Block Attention Modulecited by 2,463
  4. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
  5. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
  6. 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,352
  7. 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
  8. 2022Simple Baselines for Image Restorationcited by 701
  9. 2017FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoisingcited by 509
  10. 2018Non-Local Recurrent Network for Image Restorationcited by 160

Built on by

Not cited yet.