atlas

Research map / Image generation / Autoregressive and latent generation

XAttnMark: Learning Robust Audio Watermarking with Cross-Attention

ICML 2025 · Image generation › Autoregressive and latent generation · approach: Robust diffusion watermarking

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.

Proposes XAttnMark, a neural audio watermark with partial generator-detector parameter sharing, cross-attention message decoding, temporal message conditioning, and a psychoacoustic TF masking loss, achieving state-of-the-art detection and attribution robustness.

Problem

Neural audio watermarking has struggled to jointly achieve robust detection and accurate attribution (message decoding), especially under standard edits and generative editing.

Approach

Partially shares parameters between generator and detector via a shared message embedding table used as key/value in a cross-attention message decoder, distributes the message temporally before injection, and weights a mel-spectrogram l2 loss by a psychoacoustic per-tile masking energy computed with asymmetric TF decay.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use XAttnMark instead of AudioSealaudio watermarking · Standard edits (average)detection accuracy: 0.971 -> 0.9919Helped
use XAttnMark instead of AudiowMarkaudio watermarking · Standard edits (average)attribution accuracy: 0.88 -> 0.93Helped
use XAttnMark instead of AudioSealaudio watermarking · Stable Audio generative edits, t=10, 70, 110detection accuracy: 0.69 -> 0.94; 0.59 -> 0.91; 0.59 -> 0.91Helped
use XAttnMark instead of AudioSealaudio watermarking · AudioLDM2 generative edits, t=10, 70, 110detection accuracy: 0.59 -> 0.94; 0.59 -> 0.94; 0.59 -> 0.94Helped
use XAttnMark instead of WavMarkaudio watermarking · MusicCaps, N=100attribution accuracy: 85.08 -> 93.87Helped
add psychoacoustic TF masking loss (ablation)XAttnMark · Watermarked audio qualitySI-SNR: 19.64 -> 29.00Helped
add psychoacoustic TF masking loss (ablation)XAttnMark · Watermarked audio qualityPESQ: 4.25 -> 4.43Helped
add adaptive γ bandwidth scaling (ablation)XAttnMark · Watermarked audio qualitySI-SNR: 21.89 -> 29.00Helped
add adaptive γ bandwidth scaling (ablation)XAttnMark · Watermarked audio qualityPESQ: 4.22 -> 4.43Helped
remove temporal message modulation (ablation)XAttnMark · message decodingaccuracy: approximately 98% -> around 60%Hurt
remove cross-attention (ablation)XAttnMark · message decodingaccuracy: approximately 98% -> 50%Hurt
replace adaptive γ with constant γ=1 (ablation)XAttnMark · Watermarked audio qualitySTOI: 1.000 -> 0.991Hurt

Surprising findings

Limitations

Experimental setup

Datasets: VoxPopuli LibriSpeech MusicCaps Free Music Archive AudioSet

Benchmarks: MusicCaps held-out test set (100 files, 16 edits) VoxPopuli AudioMarkBench

Metrics: detection accuracy TPR/FPR attribution accuracy FAR SI-SNR PESQ STOI ViSQOL watermark residual loudness MUSHRA

Baselines: AudiowMark WavMark TimbreWM AudioSeal

Builds on: AudioSeal WavMark EnCodec

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