atlas

Research map / Large language models / Language, safety and interpretability

SaulLM-54B & SaulLM-141B: Scaling Up Domain Adaptation for the Legal Domain

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Adapts Mixtral-54B and Mixtral-141B to the legal domain via continued pretraining on a 500B+ token legal corpus, then legal-and-general instruction tuning and DPO alignment with synthetic legal preference data, releasing SaulLM-54B/141B under MIT.

Problem

Legal LLMs have been limited to small scale (7-12B) and small training corpora (under 30B tokens); it is unclear how much scaling model and corpus size improves legal task performance.

Approach

Scale domain adaptation of Mixtral MoE models (54B and 141B) through a full pipeline: continued pretraining on over 540B legal tokens, legal plus general instruction fine-tuning with synthetic dialogues, and DPO with synthetic legal preference pairs judged by an LLM.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Continue pretraining on legal-domain data (ablation)SaulLM models · Legal-MMLUperformance: approximately +7%Helped
Add DPO preference alignment after instruction fine-tuning (ablation)SaulLM-medium · LegalBench-Instructtask-level performanceMixed
Add legal data to the instruction fine-tuning and DPO datasets (ablation)Mixtral-54-IFT+DPO · LegalBench-Instructlegal performanceHelped
Scale from 54B to 141B (ablation)SaulLM · LegalBench-Instructtask-level performance: 20%Mixed
Scale from 54B to 141B (ablation)SaulLM · LegalBench-Instruct — international lawperformanceHurt
Continue pretraining for longer (ablation)SaulLM-54B-base and SaulLM-141B-base · Pretraining lossnormalized log lossHelped

Surprising findings

Limitations

Experimental setup

Datasets: FreeLaw MultiLegal Pile SlimPajama LawInstruct UltraChat UltraInteract Dolphin UltraFeedback Orca LegalBench-Instruct MMLU

Benchmarks: LegalBench-Instruct MMLU (international law, professional law, jurisprudence)

Metrics: balanced accuracy

Baselines: GPT-4 Llama3-70B-Instruct Mixtral-54B-Instruct Mixtral-141B-Instruct SaulLM-7B Legal-FLAN-T5

Builds on: Mixtral SaulLM-7B DPO LegalBench-Instruct

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy