トップページ
研究活動
AI for Scienceプラットフォーム部門紹介
AI学習最適化基盤開発ユニット
AI学習最適化基盤開発ユニット
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学習最適化基盤開発ユニット ユニットリーダー(現職)
キーワード
- 人工知能に基づく科学
- 基盤モデルのためのインフラストラクチャ(訓練、推論)
- 科学における生成AI
- 科学におけるAIの統合
研究概要
AI学習最適化基盤開発ユニットは、商用ユースケースを超えて科学分野に AI の使用を拡大するための AI ソフトウェア インフラストラクチャとモデルの開発に焦点を当てた R-CCS ユニットです。 特に、半構造化されたマルチモーダルデータ、複雑な表現、および帰属の科学的要件に適応します。 具体的には、以下のテーマを中心に次世代AIシステムの研究を行っています。
1. さまざまな科学分野にわたる複雑な最適化問題に対処するために、生成 AI 技術をスケーリングするための新しい最適化手法を調査および開発します。 生成 AI モデルの統合を通じて、最適化パフォーマンス、スケーラビリティ、適応性を強化することに重点を置いています。
2. 大規模言語モデル (LLM) の並列および分散推論技術を研究、開発、実装し、LLM 上に構築された効率的で科学的なアプリケーションを可能にします。 リアルタイムおよび大規模な科学アプリケーション向けに、LLM の推論速度、スケーラビリティ、およびリソース利用率を向上させることに重点を置いています。
3. 最新のハードウェア アーキテクチャの計算能力を活用し、最適化プロセスを加速するために、並列分散 DNN を研究、開発、実装します。さまざまなドメインにわたる大規模な問題に対する最適化アルゴリズムのスケーラビリティ、効率、堅牢性の向上に焦点を置いています。
主な研究成果
- 特にモデルトレーニングの効率化や科学用途向けのニューラルネットワークアーキテクチャの最適化などの分野で、機械学習のための実用的でスケーラブルなソリューションを開発する(NeurIPS’23、TEVC’24、TPDS’22、HPDC’21、SC’20)。
- 実用的な現実世界の科学問題に対する AI ソリューションの開発: 分子動力学シミュレーション (Nature Sci. Data'22)、ニューロモーフィック コンピューティング チップ シミュレーション (TPDS'23)、画像の再構成と処理 (SC'22、SC'21)、 欧州宇宙機関 (ESA) の惑星間飛行軌道 (SoftwareX'21)。
主要論文
-
Wengang Li, Lingqi Zhang, Toshio Endo, Mohamed Wahib:
"Understanding Routing Mechanism in Mixture-of-Experts Language Models."
International Conference on Learning Representations (ICLR 2026) -
Enzhi Zhang, Peng Chen, Jun Igarashi, Isaac Lyngaas, Rui Zhong, Du Wu, Xiao Wang, Masaharu Munetomo, MohamedWahib,
"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 -
Cheng Chen, Shoki Ohta, Takayuki Nishio, Mehdi Bennis, Jihong Park, Mohamed Wahib,
"Enabling Visual Scene Recovery from Wi-Fi CSI for Occlusion-Free Surveillance",
in IEEE Internet of Things Journal, vol. 12, no. 11, pp. 15040-15056, 1 June1, 2025 -
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) -
Yu Xue, Jiajie Zha, Danilo Pelusi, Peng Chen, Tao Luo, Liangli Zhen, Yan Wang, Mohamed Wahib, Neural ArchitectureSearch With
"Progressive Evaluation and Sub-Population Preservation",
in IEEE Transactions on Evolutionary Computation, vol. 29, no. 5, pp. 1678-1691, Oct. 2025 -
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) -
*Huaipeng Zhang, Nhut-Minh Ho, Yigit Polat Dogukan, Peng Chen, Mohamed Wahib, Truong Thao Nguyen, JintaoMeng, Rick Siow Mong Goh, Satoshi Matsuoka, Tao Luo, Weng-Fai Wong,
"Simeuro: A Hybrid CPU-GPU Parallel Simulator for Neuromorphic Computing Chips,"
IEEE Transactions on Parallel and Distributed Systems, volume:34, pp2767-2782 (2023) -
*Jintao Meng, Peng Chen, Mingjun Yang, Mohamed Wahib, Yanjie Wei, Shengzhong Feng, Wei Liu,Liangzhen Zheng
"Boosting the Predictive Performance with Aqueous Solubility Dataset Curation"
Nature Scientifi c Data, March 2022
メンバー
主宰者
- WAHIB Mohamed
- ユニットリーダー
メンバー
- CHEN Peng
- 上級研究員
- 五十嵐 潤
- 上級研究員
- 西口 浩司
- 上級研究員
- VATAI Emil
- 研究員
- ABEYWICKRAMA Tenindra Nadeeshan
- 研究員
- SALLINEN Scott Nathan
- 研究員
- ZHANG Lingqi
- 特別研究員
- BATISTA Joao Eduardo
- 特別研究員
- BAI Zhengyang
- 特別研究員
- 平 理一郎
- 客員研究員
