You can deploy this model and start earning money today!

sanchitahuja205/
xelm-gemma-4b-indic-freeze

Name: xelm-gemma-4b-indic-freeze
Availability: InStock

continual-pretraining

Views

No views yet

xelm-gemma-4b-indic-freeze

Layer-freezing strategy: middle transformer layers are frozen at the base Gemma-3-4B weights; only the first and last layers are updated during CPT. Mitigates catastrophic forgetting of general capabilities.

Base model: google/gemma-3-4b-pt
Strategy: freeze
Language family: Indic
Code: https://github.com/sanchit-ahuja/scaling-multilingual-experts

Loading

python

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("sanchitahuja205/xelm-gemma-4b-indic-freeze")
tokenizer = AutoTokenizer.from_pretrained("sanchitahuja205/xelm-gemma-4b-indic-freeze")

Training recipe

The exact training recipe lives in configs/yaml/train_gemma_freeze.yaml in the code repo. The resolved config used for this specific run is also included in this model repo as training_config.yaml — load it with pyrallis to reproduce the run bit-for-bit:

python train.py --config_path configs/yaml/train_gemma_freeze.yaml

Citation

bibtex

@misc{ahuja2026parameteralignmentmitigatescatastrophic,
      title={Parameter Alignment Mitigates Catastrophic Forgetting in Multilingual Expert Language Models},
      author={Sanchit Ahuja and Terra Blevins},
      year={2026},
      eprint={2606.00284},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2606.00284},
}