計算科学研究センター 高性能人工知能システム研究チーム
チームプリンシパル WAHIB Mohamed（Ph.D.）
研究概要
R-CCSの研究チームの一つである「高性能人工知能システム研究チーム」は、「富岳」などの最先端の高性能計算基盤を用いた次世代のAIの探求を行うため、国内のHPCおよびAI研究機関と連携した上で、“人工知能・機械学習と高性能計算の相互的な高性能化”を目指し、特にそのための高性能システム・ソフトウェア・アルゴリズムの研究を行います。特に、具体的には深層学習の超スケール・超高速化：深層学習の最新のアルゴリズムやフレームワークと、HPCでのMPIなどの超並列化をサポートするシステムソフトウェアを用いて、「富岳」を含む大規模なスーパーコンピュータ基盤上で超スケーリングするシステム研究、深層学習や他の機械学習・AI関連のアルゴリズムを分析し、それらの性能モデリングを行い、演算カーネル部分を最新のハードウェアアーキテクチャにおいて加速するシステム研究、超高速・超スケール化した機械学習のシステムを用いて、超深層なニューラルネットワークや、高解像度GANなどの、莫大な計算量が必要な先進的AIアルゴリズムを加速するシステム研究、機械学習に代表される経験則的手法でHPCのアルゴリズムやアプリケーションを大幅に加速するシステム研究などを、内外のHPCおよびAIに関する研究機関と連携して行っていきます。
研究主分野
- 情報学
研究関連分野
- 高性能計算
- 並列・分散処理
- 計算機システム
キーワード
- 高性能人工知能システム
- 知能プログラミングシステム
- 深層学習等の人工知能アルゴリズムの性能モデリング
- スケーラブルな深層学習
- 人工知能とシミュレーションの融合
主要論文
「*」は、理研外のみでの成果です。
- 1.
*Zhengyang Bai, Peng Chen, Mohamed Wahib,
"RT-RkNN: Reverse k Nearest Neighbor Queries as a Graphics Ray Casting Problem",
International Conference on Very Large Data Bases, Vol. 19, No. 9, (VLDB 2026) - 2.
*Chen Zhuang, Lingqi Zhang, Benjamin Brock, Du Wu, Peng Chen, Toshio Endo, Satoshi Matsuoka, Mohamed Wahib,
"SHIRO: Near-Optimal Communication Strategies for Distributed Sparse Matrix Multiplication",
ACM International Conference on Supercomputing 2026 (ICS’26) - 3.
*Wengang Li, Lingqi Zhang, Toshio Endo, Mohamed Wahib:
"Understanding Routing Mechanism in Mixture-of-Experts Language Models."
International Conference on Learning Representations (ICLR 2026) - 4.
*Lingqi Zhang, Tengfei Wang, Jiajun Huang, Chen Zhuang, Ivan R. Ivanov, Peng Chen, Toshio Endo, Mohamed Wahib,
"FRUGAL: Pushing GPU Applications Beyond Memory Limits",
Proceedings of the 24th ACM/IEEE International Symposium on Code Generation and Optimization 2026, pp 199-211(CGO'26) - 5.
*Enzhi Zhang, Peng Chen, Jun Igarashi, Isaac Lyngaas, Rui Zhong, Du Wu, Xiao Wang, Masaharu Munetomo, Mohamed Wahib,
"SHFP: Symmetrical Hierarchical Forest Patching with Pretrained Vision Transformer Encoder for High-Resolution Medical Segmentation",
Advances in Neural Information Processing Systems (NeurIPS 2025) - 6.
*Mohamed Wahib, Muhammet Abdullah Soytürk, Didem Unat,
"Balanced and Elastic End-to-end Training of Dynamic LLMs",
Proceedings of the International Conference on High Performance Computing, Networking, Storage and Analysis (SC’25) - 7.
Xiao Wang, Jong-Youl Choi, Takuya Kurihaya, Isaac Lyngaas, Hong-Jun Yoon, Nasik Muhammad Nafi, Aristeidis Tsaris, Ashwin M. Aji, Maliha Hossain, Ming Fan, Mohamed Wahib, Dali Wang, Peter Thornton, Moetasim Ashfaq, Prasanna Balaprakash, Dan Lu,
"ORBIT-2: Scaling Exascale Vision Foundation Models for Weather and Climate Downscaling",
Proceedings of the International Conference on High Performance Computing, Networking, Storage and Analysis (SC’25) Gordon Bell Finalist, Best Paper Award - 8.
Faveo Hoerold, Ivan Radanov Ivanov, Akash Dhruv, William S. Moses, Anshu Dubey, Mohamed Wahib, Jens Domke,
"RAPTOR: Numerical Profiling of Scientific Applications,Proceedings of the International Conference on High Performance Computing",
Networking, Storage and Analysis (SC’25) Best Reproducibility Advancement Award - 9.
Enzhi Zhang, Isaac Lyngaas, Peng Chen, Xiao Wang, Jun Igarashi, Yuankai Huo, Masaharu Munetomo, Mohamed Wahib
"Adaptive Patching for High-resolution Image Segmentation with Transformers"
International Conference for High Performance Computing, Networking, Storage, and Analysis (SC 2024) - 10.
Thao Nguyen Truong, Balazs Gerofi, Edgar Josafat Martinez-Noriega, Francois Trahay, Mohamed Wahib
"KAKURENBO:"Adaptively Hiding Samples in Deep Neural Network Training"
Advances in Neural Information Processing Systems 2023 (NeurIPS 2023)
関連リンク
メンバーリスト
主宰者
- WAHIB Mohamed
- チームプリンシパル
メンバー
- 五十嵐 潤
- 上級研究員
- VATAI Emil
- 研究員
- BAI Zhengyang
- 特別研究員
- ZHANG Lingqi
- 特別研究員
- BATISTA Joao Eduardo
- 特別研究員
- WU Du
- 大学院生リサーチ・アソシエイト
- ZHUANG Chen
- 大学院生リサーチ・アソシエイト
- MA Cong
- 大学院生リサーチ・アソシエイト
- GEROFI Balazs
- 客員研究員
お問い合わせ先
〒103-0027 東京都中央区日本橋1-4-1 日本橋一丁目三井ビルディング 15階
Email: mohamed.attia@riken.jp