LICENSE
README.md
pyproject.toml
slim_diff.egg-info/PKG-INFO
slim_diff.egg-info/SOURCES.txt
slim_diff.egg-info/dependency_links.txt
slim_diff.egg-info/entry_points.txt
slim_diff.egg-info/requires.txt
slim_diff.egg-info/top_level.txt
src/diffusion/__init__.py
src/diffusion/config/validation.py
src/diffusion/data/__init__.py
src/diffusion/data/caching.py
src/diffusion/data/dataset.py
src/diffusion/data/splits.py
src/diffusion/data/transforms.py
src/diffusion/data/caching/__init__.py
src/diffusion/data/caching/__main__.py
src/diffusion/data/caching/base.py
src/diffusion/data/caching/cli.py
src/diffusion/data/caching/registry.py
src/diffusion/data/caching/builders/__init__.py
src/diffusion/data/caching/builders/brats_men.py
src/diffusion/data/caching/builders/epilepsy.py
src/diffusion/data/caching/utils/__init__.py
src/diffusion/data/caching/utils/config_utils.py
src/diffusion/data/caching/utils/io_utils.py
src/diffusion/data/caching/utils/metadata.py
src/diffusion/data/caching/utils/visualization.py
src/diffusion/losses/__init__.py
src/diffusion/losses/diffusion_losses.py
src/diffusion/losses/focal_frequency_loss.py
src/diffusion/losses/uncertainty.py
src/diffusion/model/__init__.py
src/diffusion/model/factory.py
src/diffusion/model/components/__init__.py
src/diffusion/model/components/anatomical_encoder.py
src/diffusion/model/components/conditioning.py
src/diffusion/model/components/fpn_backbone.py
src/diffusion/model/components/prior_map_loader.py
src/diffusion/model/components/rotary_embedding.py
src/diffusion/model/embeddings/__init__.py
src/diffusion/model/embeddings/zpos.py
src/diffusion/model_comparison/__init__.py
src/diffusion/model_comparison/config.py
src/diffusion/model_comparison/data_loader.py
src/diffusion/model_comparison/runner.py
src/diffusion/model_comparison/utils.py
src/diffusion/model_comparison/visualizations/__init__.py
src/diffusion/model_comparison/visualizations/base.py
src/diffusion/model_comparison/visualizations/loss_curves.py
src/diffusion/model_comparison/visualizations/publication_panel.py
src/diffusion/model_comparison/visualizations/reconstruction_comparison.py
src/diffusion/model_comparison/visualizations/summary_table.py
src/diffusion/model_comparison/visualizations/timestep_mse_heatmap.py
src/diffusion/model_comparison/visualizations/uncertainty_evolution.py
src/diffusion/scripts/analyze_lesion_distribution.py
src/diffusion/scripts/analyze_lesion_reconstruction_error.py
src/diffusion/scripts/analyze_split_distribution.py
src/diffusion/scripts/check_replica_quality.py
src/diffusion/scripts/plot_image_grid.py
src/diffusion/scripts/plot_kid_results.py
src/diffusion/scripts/postprocess_replicas.py
src/diffusion/scripts/visualize_binning.py
src/diffusion/scripts/visualize_cache_bias.py
src/diffusion/scripts/visualize_patient_with_priors.py
src/diffusion/scripts/visualize_zbin_priors.py
src/diffusion/scripts/similarity_metrics/__init__.py
src/diffusion/scripts/similarity_metrics/__main__.py
src/diffusion/scripts/similarity_metrics/cli.py
src/diffusion/scripts/similarity_metrics/run_icip2026.py
src/diffusion/scripts/similarity_metrics/run_mask_metrics.py
src/diffusion/scripts/similarity_metrics/data/__init__.py
src/diffusion/scripts/similarity_metrics/data/loaders.py
src/diffusion/scripts/similarity_metrics/metrics/__init__.py
src/diffusion/scripts/similarity_metrics/metrics/feature_nn.py
src/diffusion/scripts/similarity_metrics/metrics/fid.py
src/diffusion/scripts/similarity_metrics/metrics/kid.py
src/diffusion/scripts/similarity_metrics/metrics/lpips.py
src/diffusion/scripts/similarity_metrics/metrics/mask_morphology.py
src/diffusion/scripts/similarity_metrics/plotting/__init__.py
src/diffusion/scripts/similarity_metrics/plotting/global_comparison.py
src/diffusion/scripts/similarity_metrics/plotting/icip2026_figure.py
src/diffusion/scripts/similarity_metrics/plotting/mask_comparison.py
src/diffusion/scripts/similarity_metrics/plotting/nn_comparison.py
src/diffusion/scripts/similarity_metrics/plotting/settings.py
src/diffusion/scripts/similarity_metrics/plotting/zbin_multiexp.py
src/diffusion/scripts/similarity_metrics/statistics/__init__.py
src/diffusion/scripts/similarity_metrics/statistics/comparison.py
src/diffusion/tests/__init__.py
src/diffusion/tests/test_anatomical_weights.py
src/diffusion/tests/test_csv_callback.py
src/diffusion/tests/test_enhanced_encoder.py
src/diffusion/tests/test_full_loss_scenario.py
src/diffusion/tests/test_local_binning.py
src/diffusion/tests/test_loss_configurations.py
src/diffusion/tests/test_loss_regression.py
src/diffusion/tests/test_loss_scenarios.py
src/diffusion/tests/test_multicomponent_priors.py
src/diffusion/tests/test_reconstruction_validation.py
src/diffusion/tests/test_sinusoidal_embedding.py
src/diffusion/tests/test_sinusoidal_local_binning.py
src/diffusion/tests/test_smoke.py
src/diffusion/tests/test_uncertainty_loss.py
src/diffusion/tests/test_validate_metrics.py
src/diffusion/tests/test_visualize_data.py
src/diffusion/tests/test_visualize_priors.py
src/diffusion/tests/test_visualize_zbin_priors.py
src/diffusion/tests/test_zbin_priors.py
src/diffusion/training/__init__.py
src/diffusion/training/lesion_metrics.py
src/diffusion/training/lit_modules.py
src/diffusion/training/metrics.py
src/diffusion/training/callbacks/__init__.py
src/diffusion/training/callbacks/csv_callback.py
src/diffusion/training/callbacks/diagnostics_callbacks.py
src/diffusion/training/callbacks/encoder_monitoring_callback.py
src/diffusion/training/callbacks/epoch_callbacks.py
src/diffusion/training/callbacks/step_callbacks.py
src/diffusion/training/callbacks/texture_quality_callback.py
src/diffusion/training/runners/__init__.py
src/diffusion/training/runners/generate.py
src/diffusion/training/runners/generate_from_spec.py
src/diffusion/training/runners/generate_replicas.py
src/diffusion/training/runners/train.py
src/diffusion/utils/__init__.py
src/diffusion/utils/io.py
src/diffusion/utils/logging.py
src/diffusion/utils/seeding.py
src/diffusion/utils/visualize_zbin_priors.py
src/diffusion/utils/zbin_priors.py
src/shared/__init__.py
src/shared/ablation/__init__.py
src/shared/ablation/ablation_space.py
src/shared/ablation/comparison.py
src/shared/ablation/discovery.py
src/shared/ablation/experiment_coords.py