Machine Learning Engineer
San Jose, California, United States
Explored Direct Preference Optimization (DPO), an innovative offline algorithm for aligning language models with human preferences.Successfully reproduced the original DPO paper, validating its methodologies and results.Conducted in-depth analysis of the fine-tuning process, with a focus on model forgetting phenomena.Investigated the impact of the beta hyperparameter on model performance, exploring the trade-offs between alignment and base model deviation.Quantified improvements of DPO over standard Supervised Fine-Tuning (SFT) approaches, demonstrating enhanced alignment with human preferences.Contributed to the AI community by fine-tuning and releasing an improved version of the BTLM-3B-8K model using DPO, now available on Hugging Face: https://huggingface.co/cerebras/btlm-3b-8k-chat.