atlas

Research map / Large language models / Alignment and preferences

MusicRL: Aligning Music Generation to Human Preferences

ICML 2024 · Large language models › Alignment and preferences · approach: LLM-as-judge and feedback data · 3 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.

Finetunes MusicLM with RL (REINFORCE with KL) using MuLan text-adherence, a learned acoustic quality reward, and a user-preference reward model trained on 300k AITK pairwise preferences, with the combined MusicRL-RU performing best in human side-by-side ratings.

Problem

Text-to-music models trained with next-token prediction lack a notion of musicality and text adherence, and music appreciation is subjective, so rater-based or automatic metrics only partially capture human preferences.

Approach

Apply KL-regularized REINFORCE to a MusicLM autoregressive stage using rewards from MuLan text-audio similarity, a quality MOS predictor, and a Bradley-Terry reward model trained on large-scale implicit user pairwise preferences collected via a public web interface, then combine them sequentially.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
fine-tune MusicLM on combined MuLan and quality rewardsMusicLM · text-to-music generationhuman preference win rate against MusicLM: 83%Helped
fine-tune MusicLM on user preference rewardMusicLM · text-to-music generationhuman preference win rate against MusicLM: 74%Helped
add user preference reward fine-tuning after MuLan and quality reward fine-tuningMusicRL-R · text-to-music generationhuman preference win rate against MusicLM: 87%Helped
add user preference reward fine-tuning after automatic reward fine-tuningMusicRL-R · text-to-music generationhuman preference win rate against MusicRL-R: 66%Helped
combine automatic reward fine-tuning with user preference reward fine-tuningMusicRL-RU · text-to-music generationhuman preference win rate against MusicRL-U: 62%Helped
fine-tune on quality reward aloneMusicLM · MusicLM RL fine-tuningquality score (MOS): 3.5 MOS to 4.6 MOSHelped
fine-tune on MuLan reward aloneMusicLM · MusicLM RL fine-tuningMuLan score: 0.58 to 0.71Helped
fine-tune on combined MuLan and quality rewardsMusicLM · MusicLM RL fine-tuningquality score (MOS): 3.5 MOS to 4.4 MOSHelped
fine-tune on combined MuLan and quality rewardsMusicLM · MusicLM RL fine-tuningMuLan score: 0.58 to 0.71Helped
fine-tune on user preference rewardMusicLM · MusicLM RL fine-tuningquality score (MOS): 3.5 MOS to 4 MOSHelped
fine-tune on user preference rewardMusicLM · MusicLM RL fine-tuningMuLan score: 0.58 to 0.55Hurt
drop text tokens from reward model input (ablation)User preference reward model · user preference pairwise evaluationaccuracyNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: MusicCaps AITK user preference data LaMDA-generated music descriptions

Benchmarks: internal 101-prompt qualitative side-by-side evaluation

Metrics: win rate win/(win+loss) MOS MuLan score quality score reward model accuracy

Baselines: MusicLM

Builds on: MusicLM AudioLM SoundStorm MuLan REINFORCE

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