---
license: other
license_name: unresolved-upstream
tags: [source-separation, senselab, mirror]
---

# unasdiff diffusion source priors (mirror)

Mirror of the two released priors for **Unsupervised Single-Channel Audio
Separation with Diffusion Source Priors** (Shi et al., AAAI 2026),
[arXiv:2512.07226](https://arxiv.org/abs/2512.07226).

Held so that senselab's `source_separation` unasdiff backend can pin a revision
and a content hash. Nothing here is original work of the Senseable Intelligence
Group.

## Contents

| File | What it is |
|---|---|
| `speech_source.pt` | Speech prior, trained on VCTK. `num_class = 1`; its only label is `0`. |
| `sound_source.pt` | General-sound prior, trained on FSDKaggle2018. `num_class = 50`, of which indices 0–40 are the 41 trained classes; the remaining nine slots are unused headroom. |
| `atten_unet_vctk.toml` | Upstream config for the speech prior. |
| `atten_unet_fsd.toml` | Upstream config for the sound prior. |

The two label spaces are **separate**. A label index is only meaningful against
the prior it belongs to.

Each checkpoint carries `model` and `ema` state dicts. Upstream's own loader
returns the **`ema`** copy; loading `model` instead runs but separates measurably
worse, without failing.

## Provenance

- Upstream code: <https://github.com/RunwuShi/unasdiff>, pinned at
  `5a5d70cdc94fe9d034892a1c5bc68ad1a67d2daa`.
- Weights: the authors' own Google Drive releases linked from the upstream README.

## Licence status — unresolved

The upstream repository has **no `LICENSE` file and no licence statement**, so the
default is all rights reserved. A request for an explicit licence has been opened
upstream; **this mirror stays private until it is answered.**

## Known limitation, from the authors

> The source-model-based separation approach is not well suited for same-class
> source separation (e.g. speech separation), because it lacks speaker-conditioning.

senselab exposes speech–speech separation anyway, with that caveat in the
docstring, because the alternative is a user rediscovering it by measurement.

## Citation

```bibtex
@inproceedings{shi2026unasdiff,
  title     = {Unsupervised Single-Channel Audio Separation with Diffusion Source Priors},
  booktitle = {Proc. AAAI},
  year      = {2026}
}
```