Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
The paper introduces MTLSI-Net, a novel architecture for multi-task dense prediction that uses linear attention to capture global cross-task interactions with reduced computational complexity. This is achieved through a Multi-Task Multi-scale Query Linear Fusion Block, a Semantic Token Distiller, and a Cross-Window Integrated attention Block. Experiments on NYUDv2 and PASCAL-Context show that MTLSI-Net achieves state-of-the-art performance while maintaining parameter efficiency.
Achieve state-of-the-art multi-task dense prediction performance with linear complexity, sidestepping the quadratic bottleneck of standard self-attention.
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.