arrow
返回

GRASFormer: A computational framework for gradient-regularized and entropy-stabilized multi-task transformer optimization

delete2026-02-03
delete0
PRE
AI
P
Pulkit Dwivedi *
B
Benazir Islam
DOI:10.1016/j.jocs.2026.102805delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Multi-task learning (MTL) with transformer architectures represents a computationally powerful paradigm for joint optimization across related objectives. However, the underlying optimization dynamics remain difficult to stabilize due to inter-task gradient conflicts and excessively sharp attention landscapes, which lead to high-curvature loss surfaces and degraded generalization. This paper introduces GRASFormer, a Gradient Regularized Attention Sharpness Transformer, as a unified computational framework that systematically addresses these challenges through dual regularization. GRASFormer models the interdependence among task gradients via pairwise cosine similarity and dynamically reprojects conflicting directions to achieve smoother multi-objective descent. Concurrently, an entropy-based attention regularizer is formulated to control attention sharpness, thereby ensuring numerically stable and broadly distributed focus across transformer heads. The resulting framework constitutes a generalizable computational model for stable and efficient optimization in multi-task deep learning. Comprehensive experiments on synthetic optimization landscapes and benchmark datasets, including MultiMNIST, FashionMNIST-Edge, CelebA, and SVHN-Length, demonstrate that GRASFormer achieves faster convergence, improved accuracy, and enhanced robustness compared to existing state-of-the-art MTL optimizers. Statistical analyses and ablation diagnostics further confirm the theoretical consistency and computational efficiency of the proposed approach.

期刊

J
Journal of Computational Science
IF:
3.7
论文数:
252
被引数:
0

机构

N
new jersey institute of technology
学者数:
282
论文数: 154
被引数: 0
I
Indian Institute of Information Technology
学者数:
210
论文数: 125
被引数: 41
引用论文

引用论文

Transformers in Vision: A Survey视觉中的变形金刚: 一项调查
err2022-09-13
err1.1K
errOAAI
errKhan, Salman; Naseer, Muzammal; Hayat, Munawar; Zamir, Syed Waqas; Khan, Fahad Shahbaz; Shah, Mubarak
err分享
err收藏
TTST: A Top-k Token Selective Transformer for Remote Sensing Image Super-Resolution
err2024-01-01
err66
PREAI
errXiao, Yi; Yuan, Qiangqiang; Jiang, Kui; He, Jiang; Lin, Chia-Wen; Zhang, Liangpei
err分享
err收藏
Deep CNN and LSTM Architecture-Based Approach for COVID-19 Detection
err
IF0
err2023-03-23
err0
PREAI
errSourabh Mehta; Utkarsh Rastogi; Pulkit Dwivedi
err分享
err收藏
Local-Global Temporal Difference Learning for Satellite Video Super-Resolution
err2024-04-01
err0
errOAAI
errYi Xiao; Qiangqiang Yuan; Kui Jiang; Xianyu Jin; Jiang He; Liangpei Zhang; Chia-Wen Lin
err分享
err收藏
学者 查看更多内容