arXiv:2508.21572cs.IR2025-08

用Keras 3和JAX加速新闻推荐,让小显存设备也能高效训练。

NewsReX: A More Efficient Approach to News Recommendation with Keras 3 and JAX

  • 基于Keras 3和JAX构建新框架,提升计算效率。
  • 在8GB RTX 3060 Ti上实现更快训练速度。
  • 提供可复现的训练指南与参数分析,助力研究优化。

新闻推荐研究的复现与对比日益困难,源于代码库分散、配置多样及模型资源消耗大。本文提出NewsReX,一个开源库,基于Keras 3和JAX实现现代化架构,显著提升计算效率。实验表明,NewsReX较现有实现更快速。为支持广泛研究,我们提供针对自定义数据集的训练脚本与简易指南。通过日本主流媒体尼科报告(Nikkei News)的专有日文新闻数据集验证功能有效性。NewsReX使复杂实验复现更迅速、对硬件要求更低,尤其在8GB RTX 3060 Ti等低配显卡上仍具显著提速。此外,本文还分析了文献中常被忽视的关键训练参数,包括负采样策略、训练轮次数量、随机批处理的影响等,为未来研究减少冗余计算、提供最佳实践参考。代码已开源:https://github.com/igor17400/NewsReX。

原文摘要 · Abstract (English)

Reproducing and comparing results in news recommendation research has become increasingly difficult. This is due to a fragmented ecosystem of diverse codebases, varied configurations, and mainly due to resource-intensive models. We introduce NewsReX, an open-source library designed to streamline this process. Our key contribution is a modern implementation built on Keras 3 and JAX, which provides an increase in computational efficiency. Experiments show that NewsReX is faster than current implementations. To support broader research, we provide a straightforward guide and scripts for training models on custom datasets. We validated this functionality using a proprietary Japanese news dataset from Nikkei News, a leading Japanese media corporation renowned for its comprehensive coverage of business, economic, and financial news. NewsReX makes reproducing complex experiments faster and more accessible to a wider range of hardware making sure the speed up it also achieved for less powerful GPUs, like an 8GB RTX 3060 Ti. Beyond the library, this paper offers an analysis of key training parameters often overlooked in the literature, including the effect of different negative sampling strategies, the varying number of epochs, the impact of random batching, and more. This supplementary analysis serves as a valuable reference for future research, aiming to reduce redundant computation when comparing baselines and guide best practices. Code available at https://github.com/igor17400/NewsReX.

新闻推荐Keras 3JAX高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。