据获奖团队首席、之江实验室智能超算研究中心研究员刘鑫介绍,在这项工作中,研究人员引入了一个系统的设计过程,涵盖了模拟所需的基础算法、并行算法和系统级优化方法。他们基于新一代神威超级计算机,提出近似最优的张量网络并行切分和收缩方法及混合精度算法,可高效扩展至数千万核并行规模,并提供每秒4.4百亿亿次的持续计算性能,是超算领域全世界目前已知的最高混合精度浮点计算性能。
“量子计算是后摩尔时代计算的重要增效途径,是解决大规模科学计算应用的重要手段。量子模拟器作为经典计算和量子计算的桥梁,基于经典计算机实现量子计算的模拟,对下提供量子计算机的正确性验证,对上辅助用户开展量子算法设计,是当前带噪声的量子计算机研发过程中不可或缺的工具。”刘鑫说。
中国团队获奖后,有关该团队打破“量子霸权”神话的消息刷屏。那么,什么是“量子霸权”?
“量子霸权”是一个科学术语,与国际政治无关。它指的是量子计算机在某个问题上远远超过现有的计算机,这个词由美国物理学家约翰·裴士基(John Preskill)在2012年提出。由于“霸权”这个词让许多人观感不适,现在科学家更愿意把它称为“量子优越性”。
2019年10月,谷歌在国际学术期刊《自然》上发表一篇文章,宣称其率先实现了“量子霸权”:谷歌公司研发的“悬铃木”量子计算原型机,可以在200秒内完成百万量子采样,而美国最快的“顶点”超级计算机需要一万年才能模拟完成——时间上的差异高达10亿倍。
这一次,中国超算应用团队证明,谷歌公司2019年演示的随机量子电路采样任务,基于新一代神威超算也可以在短时间内完成,打破了谷歌的“量子霸权”。
具体来看,中国的量子模拟器SWQSIM可以在304秒以内,得到百万更高保真度的关联样本,在一星期内得到同样数量的无关联样本。该软件还可在60小时内完成比“悬铃木”复杂1000多倍的量子电路模拟,实现100-400比特量子电路算法的单振幅和多振幅模拟,为未来量子计算的发展提供模拟支撑。
刘鑫介绍,最开始,研究团队想用全振幅模拟的方法来设计模拟器,但这对内存的需求太大了,后来他们将设计思路调整为基于张量网络收缩方法的单振幅模拟,即牺牲一部分时间复杂度,找到一条空间复杂度可以接受的近似最优路径。开始的性能结果并不理想,后来团队通过改进方法,计算速度提升了100多倍。
这个高平衡度的最优解法,曾让团队困扰了很久,连续组织了3次算法攻关。刘鑫记得,在每周举行的例会上,大家曾激烈地讨论甚至争论,都觉得自己是对的。最后,一位团队成员的提议让大家觉得眼前一亮,于是按照这个方向去实验和改进,最后获得了理想的结果。