SlowGuess/NeurIPS_2023_OCR huggingface dataset:
Computation Time. The inference time is as of now slightly longer than BASIS. To separate a single mixture, our method that uses  $N = 20$ , 000 iterations takes 328 seconds on average, whereas BASIS takes 284 seconds for the same number of iterations. Meanwhile, the UNet only requires one forward pass through the model and can separate a mixture in less than one second. Our implementation was not optimized for computation time, but there may be strategies to speed up our algorithm in practice, which will be a focus of future work.
Code. The code for reproducing our results can be found at https://github.com/tkj516/ score_based_source_separation and is also linked on our project webpage https:// alpha-rgs.github.io.