Feb, 2024

InterroGate: 学习共享、专精和修剪多任务学习的表示

TL;DR本研究提出了一种名为 InterroGate 的新型多任务学习架构,用于减轻任务干扰并优化推理计算效率。通过可学习的门控机制自动平衡共享和任务特定的表示,同时保持所有任务的性能。在广泛的实证评估中,我们使用卷积和基于 transformer 的骨干网络,在 CelebA、NYUD-v2 和 PASCAL-Context 上展示了最先进的多任务学习结果。