[ENH] Stabilized and epsilon scaled sinkhorn solvers#84

Merged

pbarbarant merged 37 commits intomainfrom

feat/stabilized-sinkhorn

Apr 17, 2025

Commits on Mar 12, 2025

Update pre-commit config to python 3.12
Pierre-Louis Barbarant
committed
Downgrade black language version to python 3.11 in pre-commit config
Pierre-Louis Barbarant
committed
Add stabilized Sinkhorn algorithms with sparse support and epsilon scaling
Pierre-Louis Barbarant
committed
Add unit tests for Sinkhorn solvers in test_utils.py
Pierre-Louis Barbarant
committed
Replace solver_sinkhorn with solver_sinkhorn_log in FUGWSolver and adjust logging format
Pierre-Louis Barbarant
committed
Replace solver_sinkhorn_sparse with solver_sinkhorn_log_sparse in FUGWSparseSolver
Pierre-Louis Barbarant
committed
Apply final potential transformation before computing the transport plan
Pierre-Louis Barbarant
committed
Add testing of transport plan equality
Pierre-Louis Barbarant
committed

Commits on Mar 13, 2025

Update Sinkhorn solver tests to include transport plan checks and adjust evaluation frequency
Pierre-Louis Barbarant
committed
Refactor Sinkhorn solver to replace pi_diff with err for convergence checks and update logging messages
Pierre-Louis Barbarant
committed
Remove debug print statement for error in Sinkhorn EPS scaling solver
Pierre-Louis Barbarant
committed
Enhance progress bar function to support null context when verbosity is disabled
Pierre-Louis Barbarant
committed
Refactor progress tracking in Sinkhorn solvers to include verbosity option
Pierre-Louis Barbarant
committed
Import ot only if needed in init_plan_dense
Pierre-Louis Barbarant
committed
Remove numInnerItermax parameter from solver_sinkhorn_eps_scaling function and adjust train_params unpacking
Pierre-Louis Barbarant
committed

Commits on Mar 25, 2025

Updated stabilization threshold
pbarbarant
committed
Refactor Sinkhorn solver functions to improve stability and transport plan computation
Pierre-Louis Barbarant
committed
Adjust stabilization threshold and implement sparse transport plan computation
Pierre-Louis Barbarant
committed
Adjust tolerance levels in Sinkhorn solver tests for improved accuracy
Pierre-Louis Barbarant
committed
Improve docstring formatting for progress bar function
Pierre-Louis Barbarant
committed

Commits on Mar 26, 2025

Optimize matrix multiplication in Sinkhorn solver for improved performance
Pierre-Louis Barbarant
committed

Commits on Mar 27, 2025

Add mini-docstrings to test_utils
Pierre-Louis Barbarant
committed
Remove unused parameter 'is_log' from Sinkhorn solver test cases
Pierre-Louis Barbarant
committed
Refactor Sinkhorn solver test cases to remove unused parameters and simplify function calls
Pierre-Louis Barbarant
committed
Refactor Sinkhorn solver functions to improve parameter handling and enhance readability
Pierre-Louis Barbarant
committed
Add docstring to epsilon scheduler in Sinkhorn solver for clarity
Pierre-Louis Barbarant
committed
Add docstring to epsilon scheduler in solver_sinkhorn_eps_scaling for clarity
Pierre-Louis Barbarant
committed
Change tensor dtype from float32 to float64 in epsilon scheduler for improved precision
Pierre-Louis Barbarant
committed
Add docstrings
Pierre-Louis Barbarant
committed
Refactor test parameters in sinkhorn solver tests for clarity
Pierre-Louis Barbarant
committed

Commits on Apr 1, 2025

Refactor tensor initialization to use dynamic dtype and shape in Sinkhorn solver
Pierre-Louis Barbarant
committed
Update the number of minimum iterations to 37 to be consistent with POT
Pierre-Louis Barbarant
committed
Fix logging statements in Sinkhorn solvers to remove unnecessary print function
Pierre-Louis Barbarant
committed
Update tolerance and evaluation frequency in Sinkhorn solver tests for improved accuracy
Pierre-Louis Barbarant
committed

Commits on Apr 2, 2025

Remove dense sinkhorn solvers already available in POT
Pierre-Louis Barbarant
committed
Remove tests for dense sinkhorn solvers no longer needed
Pierre-Louis Barbarant
committed
Add sparse kernel matrix computation and epsilon scaling functions
Pierre-Louis Barbarant
committed