Multi-task Code LLMs: Data Mix or Model Merge?
Recent research has advocated for deploying smaller, specialized code LLMs within agentic frameworks alongside frontier models, sparking interest in efficient strategies for multi-task learning that balance performance, resource constraints, and deployment costs.. We investigate optimal approaches for creating small, multi-task code LLMs by comparing data mixing versus model merging strategies. We conduct extensive experiments across two model families (Qwen Coder and DeepSeek Coder) at two scales (2B and 7B parameters), fine-tuning them for code generation and code summarization tasks. Our evaluation on HumanEval, MBPP, and Code-to-Test (CodeXGlue) benchmarks reveals that model merging achieves the best overall performance at larger scale across model families, retaining 96% of specialized model performance on code generation tasks while maintaining summarization capabilities. Notably, merged models can even surpass individually fine-tuned models, with our best configuration of Qwen Coder 2.5 7B model achieving 92.7% Pass@1 on HumanEval compared to 90.9% for its task-specific fine-tuned equivalent. At a smaller scale, we find instead data mixing to be a preferred strategy to obtain a capable multi-task model. We further introduce a weight analysis technique to understand how different tasks affect model parameters and their implications for merging strategies. The results suggest that careful merging and mixing strategies can effectively combine task-specific capabilities without significant performance degradation, making them ideal for resource-constrained deployment scenarios.
Tue 14 AprDisplayed time zone: Brasilia, Distrito Federal, Brazil change
14:00 - 14:50 | Code Generation and OptimizationLLM4Code at Oceania I Chair(s): Lingming Zhang University of Illinois at Urbana-Champaign | ||
14:00 10mTalk | Towards Improving in-IDE Code Completion for Driver Development LLM4Code Batuhan Raif Karagoz Advanced Micro Devices (AMD), Mahesh Jayasankar Advanced Micro Devices (AMD), Saurabh Bodhe Advanced Micro Devices (AMD), Subhayan Roy Advanced Micro Devices (AMD), Lejin Varghese Advanced Micro Devices (AMD), Max Kiehn Advanced Micro Devices (AMD), Yonas Bedasso Advanced Micro Devices | ||
14:10 10mTalk | Multi-task Code LLMs: Data Mix or Model Merge? LLM4Code Mingzhi Zhu Rensselaer Polytechnic Institute, Michele Merler IBM Research, Stacy Patterson Rensselaer Polytechnic Institute, Raju Pavuluri IBM T.J. Watson Research Center, Rahul Krishna IBM Research, Boris Sobolev Cisco | ||
14:20 10mTalk | LLM-Driven SQL Remediation: Towards Safe and Explainable Code for Automated Schema Refactoring LLM4Code Antony Seabra de Medeiros Pontifical Catholic University of Rio de Janeiro, Claudio Cavalcante Pontifical Catholic University of Rio de Janeiro, Nicolaas Ruberg BNDES, Sergio Lifschitz Pontifical Catholic University of Rio de Janeiro Link to publication | ||
14:30 10mTalk | CP-Agent: Agentic Constraint Programming LLM4Code Stefan Szeider Vienna University of Technology (TU Wien) | ||
14:40 10mTalk | Diverse LLMs vs. Vulnerabilities: Who Detects and Fixes Them Better? LLM4Code Arastoo Zibaeirad University of North Carolina at Charlotte, Marco Vieira University of North Carolina at Charlotte | ||