阿里巴巴研究型实习生-多模态模型大规模训练
任职要求
1.计算机相关专业博士或硕士,在MLSYS领域发表过相关顶会文章者优先; 2.扎实的工程能力,优良的编程风格,熟悉Python/C++语言和常用设计模式,具备复杂系统的设计开发调试能力; 3.熟悉深度学习的基础理论概念,熟悉Transformer架构,熟悉主流大语言模型、多模态模型等模型特点; 4.熟练PyTorch等领域…
工作职责
1.参与包括多模态基模型Pretrain、SFT等多个阶段的训练任务优化; 2.致力于提升不同阶段模型训练负载的极限吞吐,能够针对不同模型负载系统化的分析不同阶段耗时并提供相应的优化手段,优化手段包括但不局限于算子优化、通信优化、分布式策略优化等; 3.参与训练框架对于不同硬件的支持和优化。
1.参与包括多模态基模型Pretrain、SFT等多个阶段的训练任务优化; 2.致力于提升不同阶段模型训练负载的极限吞吐,能够针对不同模型负载系统化的分析不同阶段耗时并提供相应的优化手段,优化手段包括但不局限于算子优化、通信优化、分布式策略优化等; 3.参与训练框架对于不同硬件的支持和优化。
阿里巴巴达摩院医疗AI团队,面向多模态医疗大模型的前沿研究与应用落地,旨在构建能够理解和推理不同医疗模态信息(影像、文本、临床结构化数据等)的通用智能系统。 研究方向包括但不限于: ● 多模态医疗大模型的架构设计、训练与优化 ● 医学影像(CT/MRI/超声、病理切片等)与医学文本(病历、检验报告等)的跨模态表示学习,多模态对齐,知识图谱增强 ● 医疗领域的视觉‑语言模型(VLM)、医学版GPT及多模态融合方法 ● 面向诊断与治疗决策的推理型模型和知识增强大模型 ● 少样本/零样本医疗任务泛化、可解释性与可信AI 我们拥有: ● 海量高质量、多模态临床数据(影像+文本+结构化信息) ● 充足算力资源(GPU不限,集群规模支持超大模型训练) ● 与国内外顶级医院、知名医学专家的长期科研合作 ● 在顶级会议与期刊(Nature子刊、TMI、MICCAI、CVPR、NeurIPS等)的高影响力成果 这是一个将AI大模型与医疗健康结合、面向未来的研究机会。
项目关注多模态大模型的前沿技术问题,特别是领域应用中的泛化能力不足,模型信息融合效率低,推理能力弱等关键问题,探索具备推理能力的、多模态高效融合的领域多模态大模型,沉淀可复制的技术方法,推动其在行业和领域的落地应用。 项目包含但不限于如下关键课题: 1、多模态推理技术研究:探索多模态推理数据构建与强化学习算法,关注多模态领域模型的协同推理机制,提升领域泛化能力; 2、模型原生架构探索:探索多模态数据协同处理的通用模型结构,实现跨模态数据的高效表征学习,探索理解与生成一体化的多模态大模型技术框架,进一步提升模型对领域泛化能力。
当前,大规模模型正逐步向多模态方向发展。尤其是随着SORA、GPT-4o模型的问世,其热度达到了空前的水平。目前,业界已经发布了众多多模态模型,例如:CLIP、LLaVA、Qwen-VL、Qwen-VAL、DeepSeek-VL、AnyMAL、Claude3、GPT-4v、GPT-4o、Gemini1.5、Chameleon等。多模态模型的迅速发展,也带来了新的挑战,对训练框架的效率要求越来越高: 1、不同模态数据的异质性:不同模态(如图像、文本、音频、视频)有各自的编码器(encoder)和解码器(decoder),这对计算资源、显存和通信带宽的需求各不相同。训练框架需要能够适配不同模态的特点,实现高效调度和优化。 2、数据的多样性:处理的数据模态类型多样化(如纯文本、文本+图像、文本+音频、文本+视频+音频等),且模态内的数据也可能动态变化。例如,图像的数量和大小,音频的长度等都可能不同。 现有的训练框架和并行策略已经无法很好地满足当前多模态大模型的训练需求。因此,迫切需要对现有框架进行改进和优化,以支持多模态模型的高效训练。本项目将着眼于以下几个层面技术,以解决上述特性带来的挑战,提升多模态大模型训练效率: 1、多模态场景并行策略研究,以PP为例,当前的PP并行策略都是按layer粒度均分,但多模态间的异构性会给PP并行提出更高的要求,多模态场景下不同模态的计算、显存需求不同。需要打破layer级别的粒度,支持细粒度PP划分和调度,探寻更高效的PP并行策略。 2、动态长序列场景高效训练技术研究,多模态场景总序列长度会比llm场景长很多,并且不同模态的seq len长度还是动态,在动态长序列场景怎么进行高效训练。