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

2P-03
DCGANにおけるPyTorch Distributed Data Parallelライブラリを用いた並列処理
○冨田雄大,吉田明正(明大)
近年注目されている深層学習技術の1つにGANがある.GANはデータの特徴を学習することによって,実在しないデータを生成する.また,隠れ層に畳み込みを用いたDCGANでは,より鮮明な画像の生成が可能になっているが,GANに比べて学習時間が長いことが指摘されている.最近では複数GPUを搭載した並列システムが普及しており,PyTorchのDistributed Data Parallelライブラリの利用により,複数GPUを容易に扱うことができる.本手法では2つのネットワークを持つDCGANに対して,Distributed Data Parallelライブラリを適用して高速化を実現する.NVIDIA Quadro RTX6000上での性能評価の結果,提案手法の有効性が確認された.