トップページ
研究活動
計算の科学・計算による科学研究チーム紹介
高性能人工知能システム研究チーム
高性能人工知能システム研究チーム
English
チームプリンシパル Mohamed WAHIB
(モハマド ワヒブ)-
mohamed.attia [at] riken.jp
(拠点:東京) - Emailの[at]は@にご変更ください。
- 2012
- 北海道大学、博士
- 2012
- AICS 特別研究員
- 2017
- 産総研・東工大 実社会ビッグデータ活用 オープンイノベーションラボラトリ 主任研究員
- AICS 客員研究員
- 東京工業大学 特任研究員
- 2022
- 理化学研究所 計算科学研究センター(R-CCS) 高性能人工知能システム研究チーム チームリーダー(現職)
- 2024
- 理化学研究所 計算科学研究センター(R-CCS) AI for Scienceプラットフォーム部門 AI学習最適化基盤開発ユニット ユニットリーダー(現職)
キーワード
- 高性能人工知能システム
- 知能プログラミングシステム
- 深層学習等の人工知能アルゴリズムの性能モデリング
- スケーラブルな深層学習
- 人工知能とシミュレーションの融合
研究概要
R-CCSの研究チームの一つである「高性能人工知能システム研究チーム」は、「富岳」などの最先端の高性能計算基盤を用いた次世代のAIの探求を行うため、国内のHPCおよびAI研究機関と連携した上で、“人工知能・機械学習と高性能計算の相互的な高性能化”を目指し、特にそのための高性能システム・ソフトウェア・アルゴリズムの研究を行います。特に、具体的には深層学習の超スケール・超高速化:深層学習の最新のアルゴリズムやフレームワークと、HPCでのMPIなどの超並列化をサポートするシステムソフトウェアを用いて、ポスト京を含む大規模なスーパーコンピュータ基盤上で超スケーリングするシステム研究、深層学習や他の機械学習・AI関連のアルゴリズムを分析し、それらの性能モデリングを行い、演算カーネル部分を最新のハードウェアアーキテクチャにおいて加速するシステム研究、超高速・超スケール化した機械学習のシステムを用いて、超深層なニューラルネットワークや、高解像度GANなどの、莫大な計算量が必要な先進的AIアルゴリズムを加速するシステム研究、機械学習に代表される経験則的手法でHPCのアルゴリズムやアプリケーションを大幅に加速するシステム研究などを、内外のHPCおよびAIに関する研究機関と連携して行っていきます。
主要論文
「*」は、理研外のみでの成果です。
- *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) - *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) - *Wengang Li, Lingqi Zhang, Toshio Endo, Mohamed Wahib:
"Understanding Routing Mechanism in Mixture-of-Experts Language Models."
International Conference on Learning Representations (ICLR 2026) - *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) - *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-ResolutionMedical Segmentation",
Advances in Neural Information Processing Systems (NeurIPS 2025) - *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) - 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, PrasannaBalaprakash, 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 - Faveo Hoerold, Ivan Radanov Ivanov, Akash Dhruv, William S. Moses, Anshu Dubey, Mohamed Wahib, Jens Domke,
"RAPTOR: Numerical Profi ling of Scientifi c Applications,Proceedings of the International Conference on HighPerformance Computing",
Networking, Storage and Analysis (SC’25) Best Reproducibility Advancement Award - 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) - 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
- 客員研究員
