Md Rayhanul Masud
Author directory2026
TAME: Token Attribution and Masking for Emergent misalignment
Md Rayhanul Masud | Md Rizwan Parvez
The Proceedings of the Third Workshop on Uncertainty-Aware NLP
Md Rayhanul Masud | Md Rizwan Parvez
The Proceedings of the Third Workshop on Uncertainty-Aware NLP
Fine-tuning an aligned language model on narrow, flawed data can induce harmful behavior far outside the training domain, known as emergent misalignment (EM). Prior work has localized EM in model weights, activations, and training documents, but it remains unclear which training tokens carry the relevant fine-tuning signal. We introduce TAME (Token Attribution and Masking for Emergent misalignment), a three-stage framework: token attribution scores how strongly the fine-tuning update raises each response token’s likelihood, using forward passes through a released LoRA adapter; signal characterization finds patterns among high-attribution tokens; and causal validation tests them by attribution-guided loss masking. On released EM organisms and a 6,849-example medical-advice split, attribution is concentrated (top 5% of tokens hold 32% of the mass) and, in Llama, depleted for medical vocabulary but enriched for a register of unwarranted certainty, even after controlling for token rarity. Masking high-attribution tokens during fresh fine-tuning cuts EM 23× in Llama and 36× in Qwen, with the perplexity cost concentrated on the targeted register rather than on medical content; an equal random mask leaves EM unchanged. In Llama, the attribution pattern suggests that EM- relevant signal lies more in how confidently flawed content is expressed than in its domain vocabulary; the causal masking effect itself holds across both families.
Reward Engineering for Software Tasks: A Survey of Reinforcement Learning Approaches
Md Rayhanul Masud | Azmine Toushik Wasi | Salman Rahman | Md Rizwan Parvez
Findings of the Association for Computational Linguistics: EMNLP 2026
Md Rayhanul Masud | Azmine Toushik Wasi | Salman Rahman | Md Rizwan Parvez
Findings of the Association for Computational Linguistics: EMNLP 2026
Reinforcement learning is increasingly used for code-centric software engineering tasks, including code generation, understanding, repair, testing, and optimization, especially with the rise of large language models and autonomous agents. A core challenge in these settings is reward design. Unlike standard RL domains with clear scalar objectives, software tasks involve competing goals such as correctness, security, efficiency, and readability, which are difficult to capture with a single reward. As a result, RL-for-SE systems rely on heterogeneous signals, including compilation results, unit tests, coverage metrics, retrieval scores, and learned preferences. Yet this work remains scattered across tasks and communities. To our knowledge, this survey provides the first systematic review of reward engineering for RL in software tasks. We organize prior work by reward source, granularity, and aggregation. We then distill the findings into practical guidance, including a decision matrix, a design guide, and a reward engineering reporting standard for future RL- for-SE systems.