MTLSI-Net: A Linear Semantic Interaction Network for Parameter-Efficient Multi-Task Dense Prediction
Organizations: Harbin Institute of Technology · Peng Cheng Laboratory · Harbin Institute of Technology SuZhou Research Institute
Abstract
Multi-task dense prediction aims to perform multiple pixel-level tasks simultaneously. However, capturing global cross-task interactions remains non-trivial due to the quadratic complexity of standard self-attention on high-resolution features. To address this limitation, we propose a Multi-Task Linear Semantic Interaction Network (MTLSI-Net), which facilitates cross-task interaction through linear attention. Specifically, MTLSI-Net incorporates three key components: a Multi-Task Multi-scale Query Linear Fusion Block, which captures cross-task dependencies across multiple scales with linear complexity using a shared global context matrix; a Semantic Token Distiller that compresses redundant features into compact semantic tokens, distilling essential cross-task knowledge; and a Cross-Window Integrated attention Block that injects global semantics into local features via a dual-branch architecture, preserving both global consistency and spatial precision. These components collectively enable the network to capture comprehensive cross-task interactions at linear complexity with reduced parameters. Extensive experiments on NYUDv2 and PASCAL-Context demonstrate that MTLSI-Net achieves state-of-the-art performance, validating its effectiveness and efficiency in multi-task learning.
Figures & tables
| Method | Backbone | Seg.(mIoU ) | Depth(RMSE ) | Normal(mErr ) | Boundary(odsF ) | Params(M) |
| PAD-Net [ 24 ] | HRNet-18 | 36.70 | 0.6264 | 20.85 | 76.50 | 5.02 |
| PSD [ 35 ] | HRNet-18 | 36.69 | 0.6246 | 20.87 | 76.42 | 4.71 |
| ATRC [ 2 ] | HRNet-18 | 38.90 | 0.6010 | 20.48 | 76.34 | 5.06 |
| Ours | HRNet-18 | 42.89 | 0.5892 | 20.12 | 76.40 | 13.94 |
| MQTransformer [ 26 ] | Swin-T | 43.61 | 0.5979 | 20.05 | 76.20 | 32.91 |
| DeMT [ 27 ] | Swin-T | 46.36 | 0.5871 | 20.65 | 76.90 | 32.07 |
| Method | Seg. | Parsing | Sal. | Normal | Boundary | Params | FLOPs |
| mIoU | mIoU | maxF | mErr | odsF | M | G | |
| PAD-Net [ 24 ] | 78.01 | 67.12 | 79.21 | 14.37 | 72.60 | 330 | 773 |
| MTI-Net [ 18 ] | 78.31 | 67.40 | 84.75 | 14.67 | 73.00 | 851 | 774 |
| ATRC [ 2 ] | 77.11 | 66.84 | 81.20 | 14.23 | 72.10 | 340 | 871 |
| MQTransformer [ 26 ] | 78.93 | 67.41 | 83.58 | 14.21 | 73.90 | - | - |
| TaskExpert [ 29 ] | 80.64 | 69.42 | 84.87 | 13.56 | 73.30 | 420 | 622 |
| Dataset | scale | Seg | Depth | Normal | Boundary |
| mIoU | RMSE | mErr | odsF | ||
| NYUDv2 ( ) | 1 | 44.17 | 0.6254 | 20.51 | 76.00 |
| 3 | 44.10 | 0.5985 | 20.35 | 76.02 | |
| 5 | 43.85 | 0.6014 | 20.34 | 75.96 | |
| (1,3,5) | 46.03 | 0.5968 | 20.10 | 76.34 |
Multi-Task Learning
Multi-task learning trains models to solve related tasks together by sharing representations, parameters, or supervision.
See all 391 papers