← Search

Holakou Rahmanian

5 accepted papers

2025

COS-DPO: Conditioned One-Shot Multi-Objective Fine-Tuning Framework

UAI 2025

In LLM alignment and many other ML applications, one often faces the *Multi-Objective Fine-Tuning* (MOFT) problem, *i.e.*, fine-tuning an existing model with datasets labeled w.r.t. different objectives simultaneously. To address the challenge, we propose a *Conditioned One-Shot* fine-tuning framewo

2024

Accelerating Sinkhorn algorithm with sparse Newton iterations

ICLR 2024poster

Computing the optimal transport distance between statistical distributions is a fundamental task in machine learning. One remarkable recent advancement is entropic regularization and the Sinkhorn algorithm, which utilizes only matrix scaling and guarantees an approximated solution with near-linear r…

Cited by 5SourcePDFScholar
2024

Multi-objective Optimization via Wasserstein-Fisher-Rao Gradient Flow

AISTATS 2024poster

Multi-objective optimization (MOO) aims to optimize multiple, possibly conflicting objectives with widespread applications. We introduce a novel interacting particle method for MOO inspired by molecular dynamics simulations. Our approach combines overdamped Langevin and birth-death dynamics, incorpo…

2022

Toward Understanding Privileged Features Distillation in Learning-to-Rank

NeurIPS 2022accept

In learning-to-rank problems, a \textit{privileged feature} is one that is available during model training, but not available at test time. Such features naturally arise in merchandised recommendation systems; for instance, "user clicked this item" as a feature is predictive of "user purchased this…

Cited by 18SourcePDFScholar