跳至主要內容

我的旅程

依「GPU Training」篩選中清除
機器學習2025年3月

重構 CTR 模型的訓練管線與框架

Research Developer (Bidder Team) · Bridgewell

原本是手刻的 FM 與 LR 模型搭配 FTRL、L-BFGS optimizer,資料管線混用 C# 與 Python、以 row-based 的 pbb64 格式做轉換,耗時極長。改為 TensorFlow + GPU 訓練,淘汰 LR 改用具 cross-term 的 FM,資料管線收斂成純 Python 的 parquet-in / parquet-out。

六個月資料的完整訓練從約 1.5 個月縮短到 1 週
查看詳情