AIBench Training: Balanced Industry-Standard AI Training Benchmarking

Fei Tang,Wanling Gao,Jianfeng Zhan,Chuanxin Lan,Xu Wen,Lei Wang,Chunjie Luo,Zheng Cao,Xingwang Xiong,Zihan Jiang,Tianshu Hao,Fanda Fan,Fan Zhang,Yunyou Huang,Jianan Chen,Mengjia Du,Rui Ren,Chen Zheng,Daoyi Zheng,Haoning Tang,Kunlin Zhan,Biao Wang,Defei Kong,Minghe Yu,Chongkang Tan,Huan Li,Xinhui Tian,Yatao Li,Junchao Shao,Zhenyu Wang,Xiaoyu Wang,Jiahui Dai,Hainan Ye
DOI: https://doi.org/10.1109/ispass51385.2021.00014
2021-03-01
Abstract:Earlier-stage evaluations of a new AI architecture/system need affordable AI benchmarks. Only using a few AI component benchmarks like MLPerf alone in the other stages may lead to misleading conclusions. Moreover, the learning dynamics are not well understood, and the benchmarks' shelf-life is short. This paper proposes a balanced benchmarking methodology. We use real-world benchmarks to cover the factors space that impacts the learning dynamics to the most considerable extent. After performing an exhaustive survey on Internet service AI domains, we identify and implement nineteen representative AI tasks with state-of-the-art models. For repeatable performance ranking (RPR subset) and workload characterization (WC subset), we keep two subsets to a minimum for affordability. We contribute by far the most comprehensive AI training benchmark suite. The evaluations show: (1) AIBench Training (v1.1) outperforms MLPerf Training (v0.7) in terms of diversity and representativeness of model complexity, computational cost, convergent rate, computation, and memory access patterns, and hotspot functions; (2) Against the AIBench full benchmarks, its RPR subset shortens the benchmarking cost by 64%, while maintaining the primary workload characteristics; (3) The performance ranking shows the single-purpose AI accelerator like TPU with the optimized TensorFlow framework performs better than that of GPUs while losing the latter's general support for various AI models. The specification, source code, and performance numbers are available from the AIBench homepage https://www.benchcouncil.org/aibench-training/index.html.
What problem does this paper attempt to address?