arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.05099cs.LGcs.AIcs.DC
收藏

用分治原理提高机器学习的推理性能

Improving Inference Performance of Machine Learning with the Divide-and-Conquer Principle

Oracle Labs(甲骨文实验室)

Alex Kogan

摘要:

许多流行的机器学习模型在部署到CPU上时扩展性很差。在本文中,我们探讨了原因,并提出了一个简单的,但有效的方法的基础上,众所周知的分治原则,以解决这个问题具有重要的实际意义。给定一个推理作业,而不是使用所有可用的计算资源(即,为了运行它,这个想法是将作业分解为可以并行执行的独立部分,每个部分的核心数量取决于其预期的计算成本。我们在流行的Onnxtek框架中实现了这一想法,并通过几个用例评估了其有效性,包括著名的光学字符识别模型(PaddleOCR)和自然语言处理模型(BERT)。

英文摘要:

Many popular machine learning models scale poorly when deployed on CPUs. In this paper we explore the reasons why and propose a simple, yet effective approach based on the well-known Divide-and-Conquer Principle to tackle this problem of great practical importance. Given an inference job, instead of using all available computing resources (i.e., CPU cores) for running it, the idea is to break the job into independent parts that can be executed in parallel, each with the number of cores according to its expected computational cost. We implement this idea in the popular OnnxRuntime framework and evaluate its effectiveness with several use cases, including the well-known models for optical character recognition (PaddleOCR) and natural language processing (BERT).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑