用深度学习预测NCAA篮球赛结果,对比LSTM与Transformer效果
Forecasting NCAA Basketball Outcomes with Deep Learning: A Comparative Study of LSTM and Transformer Models
- 用LSTM和Transformer建模比赛序列,结合多种统计特征
- Transformer+BCE的AUC达0.8473,表现最佳;LSTM+Brier损失校准最准
- 适合体育数据分析、模型选型研究者参考
本研究探索先进深度学习方法,用于预测2025年NCAA一级男子组与女子组篮球锦标赛结果。基于历史比赛数据,构建两种序列模型:长短期记忆网络(LSTM)与Transformer架构。通过团队实力指标(来自广义线性模型GLM)、Elo评分、种子差异及综合数据统计等特征工程增强预测能力。采用二元交叉熵(BCE)与Brier损失函数训练模型,评估分类性能与概率校准效果。结果显示,以BCE优化的Transformer具有最优判别能力(最高AUC 0.8473),而使用Brier损失训练的LSTM在概率校准上表现最佳(最低Brier分数0.1589)。研究强调根据任务需求选择合适模型与损失函数的重要性,并提供可复现的体育预测分析框架。
原文摘要 · Abstract (English)
In this research, I explore advanced deep learning methodologies to forecast the outcomes of the 2025 NCAA Division 1 Men's and Women's Basketball tournaments. Leveraging historical NCAA game data, I implement two sophisticated sequence-based models: Long Short-Term Memory (LSTM) and Transformer architectures. The predictive power of these models is augmented through comprehensive feature engineering, including team quality metrics derived from Generalized Linear Models (GLM), Elo ratings, seed differences, and aggregated box-score statistics. To evaluate the robustness and reliability of predictions, I train each model variant using both Binary Cross-Entropy (BCE) and Brier loss functions, providing insights into classification performance and probability calibration. My comparative analysis reveals that while the Transformer architecture optimized with BCE yields superior discriminative power (highest AUC of 0.8473), the LSTM model trained with Brier loss demonstrates superior probabilistic calibration (lowest Brier score of 0.1589). These findings underscore the importance of selecting appropriate model architectures and loss functions based on the specific requirements of forecasting tasks. The detailed analytical pipeline presented here serves as a reproducible framework for future predictive modeling tasks in sports analytics and beyond.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。