Cover Image

NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspective

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Qin, Xiaohan, Wang, Xiaoxing, Liao, Ning, Yan, Junchi
Format:	Preprint
Published:	2025
Subjects:	Machine Learning Artificial Intelligence
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

Parameter-Efficient Fine-Tuning for Continual Learning: A Neural Tangent Kernel Perspective
by: Liu, Jingren, et al.
Published: (2024)

Injecting Imbalance Sensitivity for Multi-Task Learning
by: Zhou, Zhipeng, et al.
Published: (2025)

When and Why Adversarial Training Improves PINNs: A Neural Tangent Kernel Perspective
by: Cao, Yuan-dong, et al.
Published: (2026)

FineRMoE: Dimension Expansion for Finer-Grained Expert with Its Upcycling Approach
by: Liao, Ning, et al.
Published: (2026)

Fast Graph Condensation with Structure-based Neural Tangent Kernel
by: Wang, Lin, et al.
Published: (2023)

Divergence of Empirical Neural Tangent Kernel in Classification Problems
by: Yu, Zixiong, et al.
Published: (2025)

Imbalanced Gradients in RL Post-Training of Multi-Task LLMs
by: Wu, Runzhe, et al.
Published: (2025)

Differential Privacy Mechanisms in Neural Tangent Kernel Regression
by: Gu, Jiuxiang, et al.
Published: (2024)

On the Emergence of Cross-Task Linearity in the Pretraining-Finetuning Paradigm
by: Zhou, Zhanpeng, et al.
Published: (2024)

The Positivity of the Neural Tangent Kernel
by: Carvalho, Luís, et al.
Published: (2024)

When Training-Free NAS Meets Vision Transformer: A Neural Tangent Kernel Perspective
by: Zhou, Qiqi, et al.
Published: (2024)

Task Scheduling & Forgetting in Multi-Task Reinforcement Learning
by: Speckmann, Marc, et al.
Published: (2025)

Task Addition in Multi-Task Learning by Geometrical Alignment
by: Yim, Soorin, et al.
Published: (2024)

Soft Conflict-Resolution Decision Transformer for Offline Multi-Task Reinforcement Learning
by: Wang, Shudong, et al.
Published: (2025)

Proactive Gradient Conflict Mitigation in Multi-Task Learning: A Sparse Training Perspective
by: Zhang, Zhi, et al.
Published: (2024)

Beyond Losses Reweighting: Empowering Multi-Task Learning via the Generalization Perspective
by: Phan, Hoang, et al.
Published: (2022)

Task-Aware Harmony Multi-Task Decision Transformer for Offline Reinforcement Learning
by: Fan, Ziqing, et al.
Published: (2024)

Learning to Solve Combinatorial Optimization under Positive Linear Constraints via Non-Autoregressive Neural Networks
by: Wang, Runzhong, et al.
Published: (2024)

Transformers from Diffusion: A Unified Framework for Neural Message Passing
by: Wu, Qitian, et al.
Published: (2024)

Efficient Multi-Task Reinforcement Learning with Cross-Task Policy Guidance
by: He, Jinmin, et al.
Published: (2025)

Molecule Generation for Drug Design: a Graph Learning Perspective
by: Yang, Nianzu, et al.
Published: (2022)

Path-Coordinated Continual Learning with Neural Tangent Kernel-Justified Plasticity: A Theoretical Framework with Near State-of-the-Art Performance
by: Shit, Rathin Chandra
Published: (2025)

Mitigating Participation Imbalance Bias in Asynchronous Federated Learning
by: Chang, Xiangyu, et al.
Published: (2025)

Dual-Balancing for Multi-Task Learning
by: Lin, Baijiong, et al.
Published: (2023)

CALM: Consensus-Aware Localized Merging for Multi-Task Learning
by: Yan, Kunda, et al.
Published: (2025)

Projected Task-Specific Layers for Multi-Task Reinforcement Learning
by: Roberts, Josselin Somerville, et al.
Published: (2023)

Enhancing Molecular Property Prediction with Auxiliary Learning and Task-Specific Adaptation
by: Dey, Vishal, et al.
Published: (2024)

Unveiling the Mechanism of Continuous Representation Full-Waveform Inversion: A Wave Based Neural Tangent Kernel Framework
by: Chen, Ruihua, et al.
Published: (2026)

Handling Distribution Shifts on Graphs: An Invariance Perspective
by: Wu, Qitian, et al.
Published: (2022)

Multi-Task Reward Learning from Human Ratings
by: Wu, Mingkang, et al.
Published: (2025)

JTok: On Token Embedding as another Axis of Scaling Law via Joint Token Self-modulation
by: Yang, Yebin, et al.
Published: (2026)

Group Fairness in Multi-Task Reinforcement Learning
by: Song, Kefan, et al.
Published: (2025)

Poisson Process for Bayesian Optimization
by: Wang, Xiaoxing, et al.
Published: (2024)

IMOVNO+: A Regional Partitioning and Meta-Heuristic Ensemble Framework for Imbalanced Multi-Class Learning
by: Bacha, Soufiane, et al.
Published: (2026)

Efficient UAV Swarm-Based Multi-Task Federated Learning with Dynamic Task Knowledge Sharing
by: Yang, Yubo, et al.
Published: (2025)

MO-EMT-NAS: Multi-Objective Continuous Transfer of Architectural Knowledge Between Tasks from Different Datasets
by: Liao, Peng, et al.
Published: (2024)

Efficient Multi-Task Reinforcement Learning via Task-Specific Action Correction
by: Feng, Jinyuan, et al.
Published: (2024)

KernelBlaster: Continual Cross-Task CUDA Optimization via Memory-Augmented In-Context Reinforcement Learning
by: Dong, Kris Shengjun, et al.
Published: (2026)

Continual Optimization with Symmetry Teleportation for Multi-Task Learning
by: Zhou, Zhipeng, et al.
Published: (2025)

Multi-Task Reinforcement Learning Enables Parameter Scaling
by: McLean, Reginald, et al.
Published: (2025)