This repository contains a minimal example demonstrating Self-Purifying Flow Matching (SPFM) in an iPython/Jupyter notebook environment.

Self-Purifying Flow Matching (SPFM) is a principled approach to filtering unreliable data within the flow-matching framework. SPFM identifies suspicious data using the model itself during the training process, bypassing the need for pretrained models or additional modules. This technique is particularly useful when dealing with noisy labels or mislabeled samples in training datasets.

# Install required packages
pip install torch numpy matplotlib jupyter

1. Clone this repository
2. Open the Jupyter notebook:

   jupyter notebook toy_example.ipynb

3. Run the cells to see SPFM in action on a toy dataset

See toy_example.ipynb for a complete minimal example demonstrating:

• Dataset preparation with synthetic noise
• SPFM training loop implementation
• Comparison with standard flow matching
• Visualization of results

If you use this code in your research, please cite:

@article{kim2025training,
  title={Training Flow Matching Models with Reliable Labels via Self-Purification},
  author={Kim, Hyeongju and Yu, Yechan and Yi, June Young and Lee, Juheon},
  journal={arXiv preprint arXiv:2509.19091},
  year={2025}
}

• Hyeongju Kim - hyeongju@supertone.ai
• Yechan Yu - ato@supertone.ai

For questions, suggestions, or contributions, please contact the maintainers or open an issue on this repository.