# Tunable MAGMAX: Preference-Aware Model Merging for Continual Learning

Source: [Tunable MAGMAX: Preference-Aware Model Merging for Continual Learning](<https://www.sota2.com/research/paper/6a0e9173f984739ed67389bc>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2026-07-24T05:00:05.475Z

Authors: Kei Hiroshima, Kento Uchida, Shinichi Shirakawa

Published: 2026-05-20T06:50:10.000Z

## Abstract

Continual learning (CL) aims to train models sequentially on multiple tasks while mitigating catastrophic forgetting of previously learned knowledge. Recent advances in large pre-trained models (LPMs) and model merging techniques, such as MAGMAX, have demonstrated effective CL performance by combining task-specific parameters. However, existing methods primarily focus on average performance across all tasks and do not adequately address how to construct models accommodating different deployment environments or varying user preferences. This paper proposes a model merging framework, termed Tunable MAGMAX, which enables preference-aware control of task-specific performance in CL. Our method introduces a preference vector that controls the number of elements selected from each task vector during model merging, allowing us to adjust the merged model performance according to their deployment needs. We further propose a method for automatically constructing appropriate preference vectors by leveraging small amounts of target environment data and datasets from model training tasks, thereby eliminating the need for manual specification. The experimental result on CL benchmark tasks demonstrates that Tunable MAGMAX effectively controls task-wise performance and successfully adapts merged models to various target environments. The proposed Tunable MAGMAX achieves superior or comparable performance to baseline methods, making it a practical solution for deploying CL models to various environments where the preferences of each task performance differ.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Image Classification on CIFAR-100-50 (test)](<https://www.sota2.com/research/sota/image-classification-on-cifar-100-50-test.md>): Image Classification; CIFAR-100-50 (test); Accuracy (D\_tar,1): 78.

- [Image Classification on CIFAR-100 5](<https://www.sota2.com/research/sota/image-classification-on-cifar-100-5.md>): Image Classification; CIFAR-100 5; D\_tar,1: 0.85.

- [Image Classification on CIFAR-100 20](<https://www.sota2.com/research/sota/image-classification-on-cifar-100-20.md>): Image Classification; Cifar-100-20; Dtar,1 Accuracy: 81.

- [Image Classification on ImageNet R-5](<https://www.sota2.com/research/sota/image-classification-on-imagenet-r-5.md>): Image Classification; ImageNet R-5; D\_tar,1: 85.

- [Image Classification on ImageNet-R-20 (Target)](<https://www.sota2.com/research/sota/image-classification-on-imagenet-r-20-target.md>): Image Classification; ImageNet-R-20 (Target); Accuracy (D\_tar,1): 83.

- [Image Classification on ImageNet-R-50 (target)](<https://www.sota2.com/research/sota/image-classification-on-imagenet-r-50-target.md>): Image Classification; ImageNet-R-50 (target); D\_tar Metric 1: 82.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2605.20803>)

[PDF](<https://arxiv.org/pdf/2605.20803.pdf>)
