情報処理学会 第84回全国大会 会期:2022年3月3日~5日 情報処理学会 第84回全国大会 会期:2022年3月3日~5日

7J-03
Cythonを用いた畳み込みニューラルネットワークにおける並列処理
○西村太輔,吉田明正(明大)
機械学習においてはPythonプログラムによる実装が広く普及している.Pythonプログラムの高速化手法として,変数の静的型付けを行うCython,JITコンパイラライブラリであるNumba,GILを解除して並列化を行うthreadingやmultiprocessing等が挙げられる.
本稿では画像分類の教師あり学習を対象とし,静的型付けを伴うCython実装による高速化を実現する.また,Cythonには並列処理を容易に実装するためのprange特殊関数が用意されており,本研究ではprange特殊関数を用いて,畳み込みニューラルネットワークのim2col関数およびcol2im関数の並列化を行った.Intel Core i7-8700の6コア上で行った性能評価の結果,提案手法の有効性が確認された.