{"id":25559,"date":"2025-03-19T20:06:32","date_gmt":"2025-03-19T12:06:32","guid":{"rendered":"http:\/\/139.9.1.231\/?p=25559"},"modified":"2025-03-19T20:06:34","modified_gmt":"2025-03-19T12:06:34","slug":"deepspeed-ulysses","status":"publish","type":"post","link":"http:\/\/139.9.1.231\/index.php\/2025\/03\/19\/deepspeed-ulysses\/","title":{"rendered":"DeepSpeed Ulysses: \u8bad\u7ec3\u6781\u957f\u5e8f\u5217Transformer\u6a21\u578b\u7684\u7cfb\u7edf\u4f18\u5316"},"content":{"rendered":"\n<ul><li><strong><em>\u535a\u5ba2\uff1a<a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/blob\/master\/blogs\/deepspeed-ulysses\/chinese\/README.md\">https:\/\/github.com\/deepspeedai\/DeepSpeed\/blob\/master\/blogs\/deepspeed-ulysses\/chinese\/README.md<\/a><\/em><\/strong><\/li><li><strong><em>\u6559\u7a0b\uff1a<a rel=\"noreferrer noopener\" href=\"https:\/\/www.deepspeed.ai\/tutorials\/ds-sequence\/\" target=\"_blank\">https:\/\/www.deepspeed.ai\/tutorials\/ds-sequence\/<\/a><\/em><\/strong><\/li><li><strong><em>DeepSpeed Ulysses\uff1a<a href=\"https:\/\/arxiv.org\/abs\/2309.14509\">https:\/\/arxiv.org\/abs\/2309.14509<\/a><\/em><\/strong><\/li><\/ul>\n\n\n\n<p>\u4ece\u751f\u6210\u6027AI\u5230\u79d1\u7814\u6a21\u578b\uff0c\u957f\u5e8f\u5217\u8bad\u7ec3\u6b63\u5728\u53d8\u5f97\u975e\u5e38\u91cd\u8981\u3002 \u5728\u751f\u6210\u6027AI\u9886\u57df\uff0c\u4f1a\u8bdd\u5f0fAI\u3001\u957f\u6587\u6863\u6458\u8981\u548c\u89c6\u9891\u751f\u6210\u7b49\u4efb\u52a1\u90fd\u9700\u8981\u5728\u7a7a\u95f4\u548c\u65f6\u95f4\u5c42\u9762\u5bf9\u957f\u4e0a\u4e0b\u6587\u8fdb\u884c\u63a8\u7406\u3002 \u4f8b\u5982\uff0c\u591a\u6a21\u6001\u57fa\u7840\u6a21\u578b\uff0c\u5982\u540c\u65f6\u5904\u7406\u8bed\u97f3\u3001\u56fe\u50cf\u548c\u6ce2\u5f62\u7684\u6a21\u578b\uff0c\u9700\u8981\u5bf9\u5177\u6709\u6781\u957f\u5e8f\u5217\u7684\u9ad8\u7ef4\u8f93\u5165\u8fdb\u884c\u957f\u4e0a\u4e0b\u6587\u63a8\u7406\u3002 \u540c\u6837\uff0c\u7ae0\u8282\u548c\u4e66\u7c4d\u7ea7\u522b\u7684\u6458\u8981\uff08\u6570\u4e07\u751a\u81f3\u6570\u5341\u4e07\u5b57\uff09\u5728\u4f1a\u8bdd\u5f0fAI\u548c\u6458\u8981\u4efb\u52a1\u4e2d\u4e5f\u975e\u5e38\u91cd\u8981\u3002<\/p>\n\n\n\n<p>\u5bf9\u4e8e\u79d1\u5b66AI\u6765\u8bf4\uff0c\u957f\u5e8f\u5217\u540c\u6837\u81f3\u5173\u91cd\u8981\uff0c\u5b83\u4e3a\u66f4\u597d\u5730\u7406\u89e3\u7ed3\u6784\u751f\u7269\u5b66\u3001\u533b\u7597\u4fdd\u5065\u3001\u6c14\u5019\u548c\u5929\u6c14\u9884\u6d4b\u4ee5\u53ca\u5927\u5206\u5b50\u6a21\u62df\u6253\u5f00\u4e86\u5927\u95e8\u3002 \u4f8b\u5982\uff0c\u901a\u8fc7\u5728\u57fa\u56e0\u5e8f\u5217\u4e0a\u8bad\u7ec3\u5927\u578b\u8bed\u8a00\u6a21\u578b\uff0c\u6211\u4eec\u53ef\u4ee5\u521b\u5efa\u53ef\u4ee5\u4f7f\u7528\u6781\u957f\u5e8f\u5217\uff08\u4eba\u7c7b\u57fa\u56e0\u7ec4\u670964\u4ebf\u4e2a\u78b1\u57fa\u5bf9\uff09\u5b66\u4e60\u57fa\u56e0\u7ec4\u8fdb\u5316\u6a21\u5f0f\u7684\u8bed\u8a00\u6a21\u578b\u3002\u5728\u533b\u7597\u4fdd\u5065\u9886\u57df\uff0c\u4ee5\u6240\u6709\u7684\u60a3\u8005\u62a4\u7406\u8bb0\u5f55\u4e3a\u6761\u4ef6\u7684\u8bca\u65ad\u9884\u6d4b\u6a21\u578b\u9700\u8981\u6781\u957f\u5e8f\u5217\u7684\u4e0a\u4e0b\u6587\u3002<\/p>\n\n\n\n<p>\u5c3d\u7ba1\u5bf9\u4e8e\u751f\u6210\u6027AI\u548c\u79d1\u5b66AI\u6765\u8bf4\uff0c\u957f\u5e8f\u5217\u957f\u5ea6\u7684\u91cd\u8981\u6027\u9010\u6e10\u589e\u957f\uff0c\u4f46\u73b0\u6709\u7684\u5927\u578b\u6a21\u578b\u8bad\u7ec3\u7cfb\u7edf\u548c\u5e95\u5c42\u7684<strong>\u5e76\u884c\u6280\u672f\uff08\u6570\u636e\u3001\u5f20\u91cf\u3001\u6d41\u6c34\u7ebf\u3001\u5e8f\u5217\u5e76\u884c\uff09\u5e76\u4e0d\u80fd\u652f\u6301\u9ad8\u6548\u7684\u957f\u5e8f\u5217\u8bad\u7ec3<\/strong>\u3002\u73b0\u6709\u5e76\u884c\u65b9\u6cd5\u5b58\u5728\u4e24\u4e2a\u4e3b\u8981\u6311\u6218\u3002\u9996\u5148\uff0c<strong>\u73b0\u6709\u7684\u6570\u636e\u3001\u5f20\u91cf\u548c\u6d41\u6c34\u7ebf\u7b49\u5e76\u884c\u65b9\u6cd5\u65e0\u6cd5\u89e3\u51b3\u5e8f\u5217\u7ef4\u5ea6\u7684\u6269\u5c55\u95ee\u9898<\/strong>\u3002\u5176\u6b21\uff0c<strong>\u7531\u4e8e\u5185\u5b58\u901a\u4fe1\u6548\u7387\u4f4e\u4e0b\uff0c\u73b0\u6709\u7684<em>\u5e8f\u5217\u5e76\u884c<\/em>\u65b9\u6cd5\u4e0d\u591f\u9ad8\u6548\u3002\u6b64\u5916\uff0c\u73b0\u6709\u65b9\u6cd5\u7684\u6613\u7528\u6027\u4e0d\u8db3\uff0c\u9700\u8981\u8fdb\u884c\u4fb5\u5165\u6027\u548c\u590d\u6742\u6613\u51fa\u9519\u7684\u4ee3\u7801\u91cd\u6784\u3002<\/strong><\/p>\n\n\n\n<p>\u4e3a\u4e86\u89e3\u51b3\u8fd9\u4e9b\u95ee\u9898\uff0c\u6211\u4eec\u5f88\u9ad8\u5174\u5ba3\u5e03\u63a8\u51fa<em>DeepSpeed-Ulysses\uff08\u6216\u79f0\u4e3aUlysses\uff0c\u4e00\u4e2a\u975e\u5e38\u957f\u7684\u5c0f\u8bf4\uff09<\/em>\uff0c\u8fd9\u662f\u4e00\u79cd\u7b80\u5355\u3001\u6613\u7528\u4e14\u9ad8\u6548\u7684\u65b9\u6cd5\uff0c\u7528\u4e8e<strong>\u652f\u6301\u5177\u6709\u6781\u957f\u5e8f\u5217\u957f\u5ea6\u7684\u9ad8\u6548\u53ef\u6269\u5c55LLM\u8bad\u7ec3\u3002<\/strong><\/p>\n\n\n\n<p>DeepSpeed-Ulysses<strong>\u5c06\u5404\u4e2a\u6837\u672c\u5728\u5e8f\u5217\u7ef4\u5ea6\u4e0a\u5206\u5272\u7ed9\u53c2\u4e0e\u7684GPU\u3002\u7136\u540e\uff0c\u5728attention\u8ba1\u7b97\u4e4b\u524d\uff0c\u5b83\u5bf9\u5df2\u5206\u5272\u7684\u67e5\u8be2(Q)\u3001\u952e(K)\u548c\u503c(V)\u6267\u884c<em>all-to-all\u901a\u4fe1<\/em>\u64cd\u4f5c\uff0c\u4ee5\u4f7f\u6bcf\u4e2aGPU\u63a5\u6536\u5b8c\u6574\u7684\u5e8f\u5217\uff0c\u4f46\u4ec5\u7528\u4e8e\u6ce8\u610f\u529b\u5934\u7684\u975e\u91cd\u53e0\u5b50\u96c6\u3002\u8fd9\u4f7f\u5f97\u53c2\u4e0e\u7684GPU\u53ef\u4ee5\u5e76\u884c\u8ba1\u7b97\u4e0d\u540c\u7684\u6ce8\u610f\u529b\u5934\u3002\u6700\u540e\uff0cDeepSpeed-Ulysses\u8fd8\u4f7f\u7528\u53e6\u4e00\u4e2aall-to-all\u6765\u5728\u6ce8\u610f\u529b\u5934\u4e0a\u6536\u96c6\u7ed3\u679c\uff0c\u540c\u65f6\u91cd\u65b0\u5728\u5e8f\u5217\u7ef4\u5ea6\u4e0a\u8fdb\u884c\u5206\u533a\u3002<\/strong><\/p>\n\n\n\n<p>DeepSpeed-Ulysses\u53ca\u5176\u4e0e\u6b64\u535a\u5ba2\u4e00\u8d77\u53d1\u5e03\u7684\u5b9e\u73b0\u7684\u5173\u952e\u7279\u6027\u5982\u4e0b\uff1a<\/p>\n\n\n\n<ul><li>\u4e0e\u73b0\u6709\u7cfb\u7edf\u76f8\u6bd4\uff0c\u5e8f\u5217\u957f\u5ea6\u589e\u52a0\u4e86<em><strong>4\u500d<\/strong><\/em>\uff0c\u652f\u6301\u8bad\u7ec3<em><strong>\u8d85\u8fc7\u767e\u4e07\u4e2atoken<\/strong><\/em>\u7684\u5e8f\u5217\u3002<\/li><li>\u4e0e\u73b0\u6709\u7cfb\u7edf\u76f8\u6bd4\uff0c\u901a\u4fe1\u51cf\u5c11\u4e86<em><strong>\u8d85\u8fc710\u500d<\/strong><\/em>\uff0c\u5bfc\u81f4\u541e\u5410\u91cf\u63d0\u9ad8\u4e86<em><strong>\u9ad8\u8fbe2.5\u500d<\/strong><\/em>\uff0c\u5e76\u4e14\u6bcf\u4e2aGPU\u7684\u6301\u7eed\u541e\u5410\u91cf\u8d85\u8fc7175 TFlops\uff08\u8d85\u8fc7\u786c\u4ef6\u5cf0\u503c\u768454%\uff09\u3002<\/li><li>\u5b8c\u5168\u901a\u7528\u7684attention\uff1aDeepSpeed<strong>\u5e8f\u5217\u5e76\u884c<\/strong>\u652f\u6301\u5bc6\u96c6\u548c\u7a00\u758f\u7684\u6ce8\u610f\u529b\uff0c\u5e76\u53ef\u4e0e\u9ad8\u6548\u7684\u6ce8\u610f\u529b\u5b9e\u73b0\uff08\u5982FlashAttention v2\uff09\u4e00\u8d77\u5de5\u4f5c\u3002<\/li><li>\u652f\u6301\u5927\u89c4\u6a21\u6a21\u578b\u8bad\u7ec3\uff1a<strong>DeepSpeed\u5e8f\u5217\u5e76\u884c<\/strong>\u4e0d\u4ec5\u652f\u6301\u5927\u5e8f\u5217\u957f\u5ea6\uff0c\u8fd8\u53ef\u4ee5\u4e0eZeRO-3\u5e76\u7528\u652f\u6301\u5927\u6a21\u578b\u5c3a\u5bf8\u3002<\/li><li>\u6613\u4e8e\u4f7f\u7528\u548c\u8fc1\u79fb\uff0c\u6700\u5c0f\u5316\u5bf9\u73b0\u6709\u8bad\u7ec3\u6846\u67b6\u7684\u4ee3\u7801\u66f4\u6539\u8981\u6c42\u3002<\/li><\/ul>\n\n\n\n<p>\u5728\u63a5\u4e0b\u6765\u7684\u7ae0\u8282\u4e2d\uff0c\u6211\u4eec\u8be6\u7ec6\u8ba8\u8bbaDeepSpeed-Ulysses\u7684\u6838\u5fc3\u8bbe\u8ba1\u3001\u901a\u4fe1\u590d\u6742\u5ea6\u5206\u6790\u3001\u5b9e\u9a8c\u8bc4\u4f30\u4ee5\u53ca\u4e0e\u73b0\u6709\u5de5\u4f5c\u7684\u6bd4\u8f83\uff0c\u5e76\u5c55\u793a\u5176\u53ef\u7528\u6027\u548c\u4f7f\u7528\u6307\u5357\u3002<\/p>\n\n\n\n<h2>DeepSpeed-Ulysses\u7684\u6838\u5fc3\u8bbe\u8ba1<\/h2>\n\n\n\n<figure class=\"wp-block-image size-large\"><img loading=\"lazy\" width=\"1024\" height=\"491\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-25-1024x491.png\" alt=\"\" class=\"wp-image-25619\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-25-1024x491.png 1024w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-25-300x144.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-25-768x369.png 768w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-25-1536x737.png 1536w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-25.png 1546w\" sizes=\"(max-width: 1024px) 100vw, 1024px\" \/><\/figure>\n\n\n\n<p>\u56fe1\u663e\u793a\u4e86DeepSpeed-Ulysses\u7684\u6838\u5fc3\u8bbe\u8ba1\u3002\u4e0e\u5df2\u77e5\u7684Transformer\u67b6\u6784\u4e00\u6837\uff0c\u8bbe\u8ba1\u7531<em>N<\/em>\u4e2a\u8f93\u5165\u5e8f\u5217\u5728<em>P<\/em>\u4e2a\u53ef\u7528\u8bbe\u5907\u4e0a\u5206\u533a\u7ec4\u6210\u3002\u6bcf\u4e2a\u672c\u5730<em>N\/P<\/em>\u5206\u533a\u90fd\u88ab\u6295\u5f71\u5230\u67e5\u8be2\uff08Q\uff09\u3001\u952e\uff08K\uff09\u548c\u503c\uff08V\uff09\u5d4c\u5165\u4e2d\u3002\u63a5\u4e0b\u6765\uff0c(QKV) \u5d4c\u5165\u901a\u8fc7\u53c2\u4e0e\u8ba1\u7b97\u8bbe\u5907\u4e4b\u95f4\u7684\u9ad8\u5ea6\u4f18\u5316\u7684\u5168\u5bf9\u5168\u96c6\u5408\uff08all-to-all collectives\uff09\u8fdb\u884c\u5168\u5c40\u7684 QKV \u6536\u96c6\u3002\u5728\u5168\u5bf9\u5168\u96c6\u5408\u540e\uff0c\u6bcf\u4e2a\u5934\u7684\u6ce8\u610f\u529b\u8ba1\u7b97\u5f62\u5f0f\u4e3a\uff1a<\/p>\n\n\n\n<figure class=\"wp-block-image size-full is-resized\"><img loading=\"lazy\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-26.png\" alt=\"\" class=\"wp-image-25621\" width=\"276\" height=\"50\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-26.png 406w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-26-300x54.png 300w\" sizes=\"(max-width: 276px) 100vw, 276px\" \/><\/figure>\n\n\n\n<p>\u6ce8\u610f\u529b\u8ba1\u7b97\u540e\uff0c\u53e6\u4e00\u4e2a\u5168\u5bf9\u5168\u96c6\u5408\u5c06\u6ce8\u610f\u529b\u8ba1\u7b97\u7684\u8f93\u51fa\u4e0a\u4e0b\u6587\u5f20\u91cf\u8f6c\u6362\u4e3a\u5e8f\u5217(<em>N\/P<\/em>)\u5e76\u884c\uff0c\u7528\u4e8eTransformer\u6a21\u578b\u5c42\u7684\u5269\u4f59\u6a21\u5757\u4e2d\u7684\u540e\u7eed\u64cd\u4f5c\uff08MLP MatMul\u3001\u5c42\u5f52\u4e00\u5316\u7b49\uff09\u3002<\/p>\n\n\n\n<h3>\u663e\u8457\u7684\u901a\u4fe1\u91cf\u51cf\u5c11<a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#%E6%98%BE%E8%91%97%E7%9A%84%E9%80%9A%E4%BF%A1%E9%87%8F%E5%87%8F%E5%B0%91\"><\/a><\/h3>\n\n\n\n<p>DeepSpeed-Ulysses\u4e0e\u5176\u4ed6\u73b0\u6709\u7684\u957f\u5e8f\u5217\u65b9\u6cd5\u7684\u533a\u522b\u5728\u4e8e\u5176\u66f4\u5c0f\u7684\u7d2f\u79ef\u901a\u4fe1\u91cf\u4ee5\u53ca\u968f\u7740\u5e8f\u5217\u5e76\u884c\u5ea6\u589e\u52a0\u800c\u66f4\u597d\u7684\u53ef\u6269\u5c55\u6027\uff0c\u5982\u4e0b\u6240\u793a\uff1a<\/p>\n\n\n\n<p>\u5728\u5177\u6709\u8282\u70b9\u5185NVSwitch\u4e92\u8fde\u548c\u8282\u70b9\u95f4\u80d6\u6811IB\u62d3\u6251\u7684\u73b0\u4ee3\u96c6\u7fa4\u4e0a\uff0c\u9488\u5bf9\u4e00\u4e2a\u805a\u5408\u6d88\u606f\u5927\u5c0f\u4e3a<em>M<\/em>\u7684\u5168\u5bf9\u5168\u4f20\u8f93\uff0c\u4f20\u8f93\u5230<em>P<\/em>\u4e2aGPU\u4e0a\u7684\u6bcf\u4e2a\u94fe\u63a5\u7684\u901a\u4fe1\u91cf\u4e3a<em>M\/P<\/em>\u3002 \u5bf9\u4e8e\u9690\u85cf\u5c42\u5927\u5c0f\u4e3ah\u3001\u5e8f\u5217\u957f\u5ea6\u4e3aN\u4e14\u5e76\u884c\u5ea6\u4e3aP\u7684Transformer\u6a21\u578b\uff0cDeepSpeed\u5e8f\u5217\u5e76\u884c\u4f1a\u5728\u6ce8\u610f\u8ba1\u7b97\u4e4b\u524d\u5bf9QKV\u6295\u5f71\u6267\u884c\u805a\u5408\u6d88\u606f\u5927\u5c0f\u4e3a<em>3Nh<\/em>\u7684\u5168\u5bf9\u5168\u64cd\u4f5c\uff0c\u5e76\u5728\u6ce8\u610f\u8ba1\u7b97\u4e4b\u540e\u5bf9\u8f93\u51fa\u4e0a\u4e0b\u6587\u6295\u5f71\u6267\u884c\u5927\u5c0f\u4e3a<em>Nh<\/em>\u7684\u53e6\u4e00\u4e2a\u5168\u5bf9\u5168\u64cd\u4f5c\u3002\u56e0\u6b64\uff0cDeepSpeed\u5e8f\u5217\u5e76\u884c\u6bcf\u4e2a\u94fe\u63a5\u7684\u805a\u5408\u901a\u4fe1\u91cf\u4e3a<em><strong>4Nh\/P\uff08\u6216O(N\/P)\u590d\u6742\u5ea6\uff09<\/strong><\/em>\u3002\u503c\u5f97\u6ce8\u610f\u7684\u662f\uff0c\u5f53N\u548cP\u6210\u6bd4\u4f8b\u589e\u52a0\u65f6\uff0c\u8fd9\u4e2a\u901a\u4fe1\u91cf\u662f\u6052\u5b9a\u7684\u3002<\/p>\n\n\n\n<p>\u76f8\u6bd4\u4e4b\u4e0b\uff0c\u73b0\u6709\u7684\u65b9\u6cd5\uff0c\u5982Megatron-LM\uff0c\u5728N\u7ebf\u6027\u589e\u957f\u7684\u60c5\u51b5\u4e0b\u4f1a\u5bfc\u81f4\u901a\u4fe1\u91cf\u7ebf\u6027\u589e\u52a0\uff0c\u800c\u4e0eP\u65e0\u5173\uff0c\u4ece\u800c\u5bfc\u81f4<em><strong>O(N)\u7684\u901a\u4fe1\u590d\u6742\u5ea6<\/strong><\/em>\u3002\u4f8b\u5982\uff0cMegatron-LM\u5bf9\u6bcf\u4e2aTransformer\u6a21\u578b\u5c42\u90fd\u6267\u884c\u4e24\u4e2a\u5927\u5c0f\u4e3a<em>Nh<\/em>\u7684<em>all-gather<\/em>\u64cd\u4f5c\uff0c\u4ee5\u53ca\u4e24\u4e2a\u5927\u5c0f\u4e3a<em>Nh<\/em>\u7684<em>reduce-scatter<\/em>\u64cd\u4f5c\u3002\u7136\u800c\uff0c\u5f53<em>P &gt;&gt; 1<\/em>\u65f6\uff0c\u5927\u5c0f\u4e3aM\u7684\u6bcf\u4e2aall-gather\u548creduce-scatter\u7684\u6210\u672c\u4ecd\u7136\u662fM\uff0c\u800c\u4e0d\u662f<em>M\/P<\/em>\u3002\u56e0\u6b64\uff0cMegatron-LM\u5e8f\u5217\u5e76\u884c\u4f1a\u5bfc\u81f4\u6bcf\u4e2a\u94fe\u63a5\u7684\u901a\u4fe1\u91cf\u4e3a<em><strong>4Nh<\/strong><\/em>\uff0c\u8fd9\u6bd4DeepSpeed\u5e8f\u5217\u5e76\u884c\u5927P\u500d\u3002\u8fd9\u4f7f\u5f97DeepSpeed\u5e8f\u5217\u5e76\u884c\u53ef\u4ee5\u5728\u5b9e\u73b0\u663e\u8457\u66f4\u9ad8\u7684\u8bad\u7ec3\u6548\u7387\u7684\u540c\u65f6\u652f\u6301\u6781\u957f\u5e8f\u5217\u8bad\u7ec3\u3002\u6211\u4eec\u7684\u5b9e\u9a8c\u8bc4\u4f30\u7ed3\u679c\u4e0e\u6b64\u7406\u8bba\u5206\u6790\u76f8\u7b26\u3002<\/p>\n\n\n\n<h3>DeepSpeed-Ulysses\u7684\u5176\u4ed6\u4eae\u70b9<a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#deepspeed-ulysses%E7%9A%84%E5%85%B6%E4%BB%96%E4%BA%AE%E7%82%B9\"><\/a><\/h3>\n\n\n\n<p><em><strong>\u901a\u7528\u7684\u6ce8\u610f\u529b\u89e3\u51b3\u65b9\u6848<\/strong><\/em><\/p>\n\n\n\n<p>DeepSpeed\u5206\u5e03\u5f0f\u6ce8\u610f\u529b\u6a21\u5757\u7684\u5b9e\u73b0\u8db3\u591f\u901a\u7528\uff0c\u4ee5\u652f\u6301\u4efb\u4f55\u7c7b\u578b\u7684\u6ce8\u610f\u529b\uff0c\u4f8b\u5982\u81ea\u6ce8\u610f\u3001\u4ea4\u53c9\u6ce8\u610f\u548c\u56e0\u679c\u6ce8\u610f\uff0c\u65e0\u8bba\u662f\u5b83\u4eec\u7684\u5bc6\u96c6\u8fd8\u662f\u7a00\u758f\u7248\u672c\uff0c\u4ee5\u53ca\u652f\u6301\u5c40\u90e8\u6ce8\u610f\u5c42\u7ea7\u4e0a\u7684\u957f\u5e8f\u5217\u7684\u5404\u79cd\u4f18\u5316\u5185\u6838\uff0c\u4f8b\u5982\u4e0d\u540c\u7248\u672c\u7684FlashAttention\u3002<\/p>\n\n\n\n<p>DeepSpeed-Ulysses\u7684\u901a\u7528\u6027\u6765\u81ea\u5176\u6838\u5fc3\u8bbe\u8ba1\u7684\u6a21\u5757\u5316\u6027\u8d28\uff1a\u4e00\u4e2a\u4ee5\u6ce8\u610f\u529b\u4e3a\u4e2d\u5fc3\u7684\u5e8f\u5217\u5e76\u884c\u8bbe\u8ba1\u3002\u5728\u6ce8\u610f\u529b\u8ba1\u7b97\u4e4b\u524d\uff0c\u5e8f\u5217\u5e76\u884c\u6027\u662f\u5bf9N\/P\u5206\u533a\u7684\uff0c\u800c\u6ce8\u610f\u529b\u8ba1\u7b97\u662f\u5bf9\u6bcf\u4e2a\u5934\u7684\u5e76\u884c\u6027\uff0c\u6bcf\u4e2a\u5934\u7684\u6ce8\u610f\u529b\u5168\u90fd\u4fdd\u7559\uff0c\u4f46\u5934\u7684\u6570\u91cf\u8f83\u5c11\uff0c\u56e0\u6b64\u6ce8\u610f\u529b\u8ba1\u7b97\u53ef\u4ee5\u7528\u4efb\u4f55\u7c7b\u578b\u7684\u6ce8\u610f\u529b\u673a\u5236\u66ff\u4ee3\uff0c\u4f8b\u5982\u5bc6\u96c6\u6ce8\u610f\u529b\u548c\u5404\u79cd\u5f62\u5f0f\u7684\u7a00\u758f\u6ce8\u610f\u529b\u3002<\/p>\n\n\n\n<p><em><strong>\u901a\u8fc7ZeRO-3\u96c6\u6210\u5b9e\u73b0\u66f4\u5927\u7684\u6a21\u578b\u548c\u66f4\u957f\u7684\u5e8f\u5217\u8bad\u7ec3<\/strong><\/em><\/p>\n\n\n\n<p>\u5c3d\u7ba1DeepSpeed\u5e8f\u5217\u5e76\u884c\u5728\u4f7f\u7528\u66f4\u957f\u7684\u5e8f\u5217\u8fdb\u884c\u8bad\u7ec3\u65f6\u51cf\u5c11\u4e86\u6fc0\u6d3b\u5185\u5b58\u7684\u4f7f\u7528\uff0c\u4f46\u5e76\u4e0d\u5f71\u54cd\u6a21\u578b\u72b6\u6001\u7684\u5185\u5b58\u5360\u7528\u3002\u56e0\u6b64\uff0c\u4e3a\u4e86\u652f\u6301\u5177\u6709\u5927\u5e8f\u5217\u957f\u5ea6\u7684\u5927\u8bed\u8a00\u6a21\u578b\u8bad\u7ec3\uff0c\u6211\u4eec\u5b9e\u73b0\u4e86DeepSpeed\u5e8f\u5217\u5e76\u884c\u4e0eZeRO-3\u7684\u96c6\u6210\u3002<\/p>\n\n\n\n<p><a href=\"https:\/\/www.microsoft.com\/en-us\/research\/blog\/zero-deepspeed-new-system-optimizations-enable-training-models-with-over-100-billion-parameters\/\">ZeRO Redundancy Optimizer Stage 3 (ZeRO-3)<\/a>&nbsp;\u662f\u4e00\u79cd\u7528\u4e8e\u8bad\u7ec3\u5927\u6a21\u578b\u7684\u5185\u5b58\u4f18\u5316\u6280\u672f\u3002\u4e0e\u4f20\u7edf\u7684\u795e\u7ecf\u7f51\u7edc\u6570\u636e\u5e76\u884c\u8bad\u7ec3\u4e2d\uff0c\u6a21\u578b\u72b6\u6001\u5728\u6570\u636e\u5e76\u884c\u7b49\u7ea7\u4e0a\u8fdb\u884c\u590d\u5236\u4e0d\u540c\uff0cZeRO-3\u901a\u8fc7\u5c06\u6a21\u578b\u72b6\u6001\u5728\u6570\u636e\u5e76\u884c\u7b49\u7ea7\u4e4b\u95f4\u8fdb\u884c\u5206\u533a\u6765\u4f18\u5316\u5185\u5b58\u4f7f\u7528\u3002\u7136\u800c\uff0c\u4f7f\u7528\u5e8f\u5217\u5e76\u884c\u65f6\uff0c\u8bad\u7ec3\u6570\u636e\u53ef\u4ee5\u5728\u6279\uff08\u6837\u672c\uff09\u548c\u5e8f\u5217\u7ef4\u5ea6\u4e0a\u8003\u8651\uff0c\u76f8\u5173\u7684\u5e76\u884c\u7fa4\u7ec4\u53ef\u4ee5\u7ec4\u5408\u6210\u4e00\u4e2a\u66f4\u5927\u7684\u7fa4\u7ec4\u4ee5\u5b9e\u73b0ZeRO\u5e76\u884c\u3002<\/p>\n\n\n\n<p>\u56e0\u6b64\uff0c\u6211\u4eec\u5c06ZeRO-3\u5206\u533a\u6269\u5c55\u5230\u6570\u636e\u5e76\u884c\u548c\u5e8f\u5217\u5e76\u884c\u7b49\u7ea7\u7684\u7ec4\u5408\u3002\u6362\u53e5\u8bdd\u8bf4\uff0c\u5728DeepSpeed\u5e8f\u5217\u5e76\u884c\u4e2d\uff0cZeRO\u5c06\u6a21\u578b\u72b6\u6001\u5206\u533a\u5728\u5e8f\u5217\u548c\u6570\u636e\u5e76\u884c\u7ec4\u4e4b\u95f4\uff0c\u5e76\u5728\u9700\u8981\u65f6\u6536\u96c6\u6bcf\u4e2a\u7b49\u7ea7\u5206\u533a\uff08allgather\uff09\u3002\u7c7b\u4f3c\u5730\uff0c\u68af\u5ea6\u5c06\u5728\u6570\u636e\u5e76\u884c\u548c\u5e8f\u5217\u5e76\u884c\u7b49\u7ea7\u4e4b\u95f4\u8fdb\u884c\u51cf\u5c11\uff0c\u7528\u4e8e\u53c2\u6570\u66f4\u65b0\u3002ZeRO\u53ef\u4ee5\u5728\u5e8f\u5217\u548c\u6570\u636e\u7ef4\u5ea6\u4e0a\u5b9e\u73b0\u5de8\u5927\u7684\u5185\u5b58\u8282\u7701\uff0c\u5e76\u4e14\u4e0d\u4ec5\u53ef\u4ee5\u6269\u5c55\u5230\u5927\u5e8f\u5217\u957f\u5ea6\uff0c\u8fd8\u53ef\u4ee5\u6269\u5c55\u5230\u5927\u6a21\u578b\u3002<\/p>\n\n\n\n<h2>\u8bc4\u4f30<a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#%E8%AF%84%E4%BC%B0\"><\/a><\/h2>\n\n\n\n<p>\u6211\u4eec\u5728GPT\uff08\u7528\u4e8e\u8bb8\u591aNLP\u4efb\u52a1\u7684\u57fa\u7840\u6a21\u578b\uff09\u4e0a\u4f7f\u7528\u6700\u591a64\u4e2aA100 GPU\uff0840GB\u663e\u5b58\uff09\u5bf9DeepSpeed-Ulysses\u8fdb\u884c\u4e86\u8bc4\u4f30\u3002\u6211\u4eec\u7684\u8bc4\u4f30\u5206\u4e3a\u56db\u4e2a\u65b9\u9762\uff1ai) \u5e8f\u5217\u957f\u5ea6\u53ef\u6269\u5c55\u6027\uff0cii) \u5bc6\u96c6\u6ce8\u610f\u529b\u7684\u541e\u5410\u91cf\u4ee5\u53ca\u4e0e\u73b0\u6709\u7cfb\u7edf\u7684\u6bd4\u8f83\uff0ciii) \u7a00\u758f\u6ce8\u610f\u529b\u7684\u541e\u5410\u91cf\u4ee5\u53ca\u4e0e\u73b0\u6709\u7cfb\u7edf\u7684\u6bd4\u8f83\uff0civ) DeepSpeed\u5e8f\u5217\u5e76\u884c\u7684\u6536\u655b\u6027\u7814\u7a76\u3002\u63a5\u4e0b\u6765\uff0c\u6211\u4eec\u5c06\u5bf9\u6bcf\u4e2a\u7c7b\u522b\u8ba8\u8bba\u548c\u5c55\u793a\u8bc4\u4f30\u7ed3\u679c\u3002<\/p>\n\n\n\n<h3>\u5e8f\u5217\u957f\u5ea6\u53ef\u6269\u5c55\u6027<a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#%E5%BA%8F%E5%88%97%E9%95%BF%E5%BA%A6%E5%8F%AF%E6%89%A9%E5%B1%95%E6%80%A7\"><\/a><\/h3>\n\n\n\n<p>\u7b2c\u4e00\u7ec4\u5b9e\u9a8c\u662f\u572812\u4ebf\u53c2\u6570\u7684GPT\u6a21\u578b\u4e0a\u5c06\u5e8f\u5217\u957f\u5ea6\u6269\u5c55\u5230100\u4e07token\u3002\u8fd9\u4e2a\u8bc4\u4f30\u7684\u7ed3\u679c\u5982\u56fe2\u6240\u793a\u3002DeepSpeed\u5e8f\u5217\u5e76\u884c\u5141\u8bb8\u968f\u7740GPU\u6570\u91cf\u7684\u589e\u52a0\u7ebf\u6027\u589e\u52a0\u5e8f\u5217\u957f\u5ea6\uff0c\u5e76\u4e14\u5e8f\u5217\u957f\u5ea6\u4e0eGPU\u6570\u91cf\u4fdd\u6301\u7ebf\u6027\u6bd4\u4f8b\u5173\u7cfb\uff0c\u9002\u5f53\u7684GPU\u6570\u91cf\u4e0b\u4fdd\u6301\u76f8\u4f3c\u7684\u8ba1\u7b97\u541e\u5410\u91cf\u3002<\/p>\n\n\n\n<figure class=\"wp-block-image size-full is-resized\"><img loading=\"lazy\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-27.png\" alt=\"\" class=\"wp-image-25623\" width=\"507\" height=\"372\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-27.png 841w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-27-300x220.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-27-768x564.png 768w\" sizes=\"(max-width: 507px) 100vw, 507px\" \/><\/figure>\n\n\n\n<h3>\u5bc6\u96c6\u6ce8\u610f\u529b\u8bc4\u4f30<\/h3>\n\n\n\n<figure class=\"wp-block-image size-full is-resized\"><img loading=\"lazy\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-28.png\" alt=\"\" class=\"wp-image-25624\" width=\"456\" height=\"330\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-28.png 865w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-28-300x217.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-28-768x557.png 768w\" sizes=\"(max-width: 456px) 100vw, 456px\" \/><\/figure>\n\n\n\n<p><a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#%E5%AF%86%E9%9B%86%E6%B3%A8%E6%84%8F%E5%8A%9B%E8%AF%84%E4%BC%B0\"><\/a><\/p>\n\n\n\n<p>\u63a5\u4e0b\u6765\uff0c\u6211\u4eec\u5728300\u4ebf\u53c2\u6570\u7684\u5bc6\u96c6\u6ce8\u610f\u529b\u6a21\u578b\u4e0a\u5bf9DeepSpeed\u5e8f\u5217\u5e76\u884c\u8fdb\u884c\u4e86\u8bc4\u4f30\uff0c\u5e76\u4e0eMegatron\u5e8f\u5217\u5e76\u884c\u572864\u4e2aA100 GPU\u4e0a\u8fdb\u884c\u4e86\u5bf9\u6bd4\u3002\u8fd9\u4e9b\u8bc4\u4f30\u7684\u7ed3\u679c\u5982\u56fe3\u6240\u793a\u3002<\/p>\n\n\n\n<p>\u6211\u4eec\u5c06DeepSpeed\u5e8f\u5217\u5e76\u884c\u4e0eMegatron-LM\u5728\u4e0d\u540c\u5e8f\u5217\u957f\u5ea6\u4e0b\u7684\u6027\u80fd\u8fdb\u884c\u4e86\u6bd4\u8f83\u3002\u5bf9\u4e8e\u6211\u4eec\u7684\u8bc4\u4f30\uff0c\u6211\u4eec\u9009\u62e9\u4e86\u80fd\u4f7fDeepSpeed\u5e8f\u5217\u5e76\u884c\u548cMegatron-LM\u5206\u522b\u8fbe\u5230\u6700\u4f73\u6027\u80fd\uff08\u901a\u8fc7\u541e\u5410\u91cf\u6216TFLOPs\u8861\u91cf\uff09\u7684\u5e8f\u5217\u957f\u5ea6-\u6279\u5927\u5c0f\u7ec4\u5408\uff0c\u6211\u4eec\u79f0\u4e4b\u4e3a\u6700\u4f73\uff08\u6279\u5927\u5c0f-\u5e8f\u5217\u957f\u5ea6\uff09\u914d\u7f6e\u3002\u5bf9\u4e8eDeepSpeed\u5e8f\u5217\u5e76\u884c\uff0c\u6211\u4eec\u59cb\u7ec8\u4f7f\u752864\u7684ZeRO\u5e76\u884c\u5ea6\u3002<\/p>\n\n\n\n<p>\u56fe3\u663e\u793a\uff0cDeepSpeed\u5e8f\u5217\u5e76\u884c\u5728\u76f8\u540c\u5e8f\u5217\u957f\u5ea6\u4e0b\u59cb\u7ec8\u4f18\u4e8eMegatron-LM\u3002\u6b64\u5916\uff0cDeepSpeed\u5e8f\u5217\u5e76\u884c\u53ef\u4ee5\u8fd0\u884c\u6bd4Megatron-LM\u66f4\u957f\u7684\u5e8f\u5217\u3002DeepSpeed\u5e8f\u5217\u5e76\u884c\u7684\u6027\u80fd\u4f18\u52bf\u5728\u4e8e\u4e24\u4e2a\u65b9\u9762\uff1a\uff081\uff09DeepSpeed\u5e8f\u5217\u5e76\u884c\u7ed3\u5408ZeRO-3\u7684\u5185\u5b58\u4f18\u5316\uff0c\u53ef\u4ee5\u5bb9\u7eb3\u66f4\u591a\u7684\u6837\u672c\uff0c\u4ece\u800c\u63d0\u9ad8\u541e\u5410\u91cf\uff1b\uff082\uff09\u76f8\u5bf9\u4e8eMegatron-LM\u5e8f\u5217\u5e76\u884c\u4e2d\u5e94\u7528\u7684<em>all-gather<\/em>\u901a\u4fe1\uff0cDeepSpeed\u5e8f\u5217\u5e76\u884c\u4f7f\u7528\u66f4\u9ad8\u6548\u7684\u5168\u5bf9\u5168\u901a\u4fe1\u3002<a target=\"_blank\" rel=\"noreferrer noopener\" href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/blob\/master\/blogs\/deepspeed-ulysses\/media\/fig3Ulysses.png\"><\/a><\/p>\n\n\n\n<p><em>\u56fe3\uff1aDeepSpeed\u548cMegatron LM\u5e8f\u5217\u5e76\u884c\u5728300\u4ebf\u53c2\u6570\u6a21\u578b\u4e0a\u7684\u5bc6\u96c6\u6ce8\u610f\u529b\u8bc4\u4f30\u3002<\/em><\/p>\n\n\n\n<h3>\u7a00\u758f\u6ce8\u610f\u529b\u8bc4\u4f30<\/h3>\n\n\n\n<figure class=\"wp-block-image size-full\"><img loading=\"lazy\" width=\"898\" height=\"619\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-29.png\" alt=\"\" class=\"wp-image-25625\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-29.png 898w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-29-300x207.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-29-768x529.png 768w\" sizes=\"(max-width: 898px) 100vw, 898px\" \/><\/figure>\n\n\n\n<p><a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#%E7%A8%80%E7%96%8F%E6%B3%A8%E6%84%8F%E5%8A%9B%E8%AF%84%E4%BC%B0\"><\/a><\/p>\n\n\n\n<p>\u7c7b\u4f3c\u5730\uff0c\u6211\u4eec\u5728300\u4ebf\u53c2\u6570\u7684\u7a00\u758f\u6ce8\u610f\u529b\u6a21\u578b\u4e0a\u5bf9DeepSpeed\u5e8f\u5217\u5e76\u884c\u8fdb\u884c\u4e86\u8bc4\u4f30\uff0c\u5e76\u4e0eMegatron\u5e8f\u5217\u5e76\u884c\u8fdb\u884c\u4e86\u5bf9\u6bd4\u3002\u6211\u4eec\u7684\u8bc4\u4f30\u7ed3\u679c\u5982\u56fe4\u6240\u793a\u3002\u7a00\u758f\u6ce8\u610f\u529b\u7684\u5b9e\u9a8c\u7ed3\u679c\u4e0e\u5bc6\u96c6\u6ce8\u610f\u529b\u5b9e\u9a8c\u7c7b\u4f3c\u3002\u6211\u4eec\u89c2\u5bdf\u5230DeepSpeed\u5e8f\u5217\u5e76\u884c\u7684\u541e\u5410\u91cf\u6027\u80fd\u76f8\u5bf9\u4e8eMegatron-LM\u63d0\u9ad8\u4e862\u500d\u4ee5\u4e0a\u3002\u901a\u8fc7\u8282\u7701\u5185\u5b58\uff0cDeepSpeed\u5e8f\u5217\u5e76\u884c\u7ed3\u5408ZeRO-3\u53ef\u4ee5\u6269\u5c55\u5230\u6bd4Megatron-LM\u66f4\u957f4\u500d\u7684\u5e8f\u5217\u957f\u5ea6\u3002<\/p>\n\n\n\n<p>DeepSpeed\u5e8f\u5217\u5e76\u884c\u5728\u76f8\u540c\u5e8f\u5217\u957f\u5ea6\u4e0b\u59cb\u7ec8\u4f18\u4e8eMegatron-LM\u3002\u4e8b\u5b9e\u4e0a\uff0c\u5f53\u524d\u7684DeepSpeed\u541e\u5410\u91cf\u53d7\u5230\u672c\u5730\u7a00\u758f\u6ce8\u610f\u529b\u5b9e\u73b0\u7684\u74f6\u9888\uff0c\u56e0\u6b64DeepSpeed\u541e\u5410\u91cf\u968f\u7740\u5e8f\u5217\u957f\u5ea6\u7684\u589e\u52a0\u800c\u964d\u4f4e\u3002\u6211\u4eec\u9884\u8ba1\uff0c\u968f\u7740\u672a\u6765\u5c40\u90e8\u7a00\u758f\u6ce8\u610f\u529b\u5b9e\u73b0\u6027\u80fd\u7684\u6539\u5584\uff0cDeepSpeed\u4e0eMegatron\u4e4b\u95f4\u7684\u6027\u80fd\u5dee\u8ddd\u5c06\u5728\u66f4\u5927\u7684\u5e8f\u5217\u957f\u5ea6\u4e0b\u8fdb\u4e00\u6b65\u589e\u52a0\u3002<a target=\"_blank\" rel=\"noreferrer noopener\" href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/blob\/master\/blogs\/deepspeed-ulysses\/media\/fig4Ulysses.png\"><\/a><\/p>\n\n\n\n<p><em>\u56fe4\uff1aDeepSpeed\u548cMegatron LM\u5e8f\u5217\u5e76\u884c\u5728300\u4ebf\u53c2\u6570\u6a21\u578b\u4e0a\u7684\u7a00\u758f\u6ce8\u610f\u529b\u8bc4\u4f30\u3002<\/em><\/p>\n\n\n\n<h3>\u6536\u655b\u6027\u7814\u7a76<\/h3>\n\n\n\n<figure class=\"wp-block-image size-full is-resized\"><img loading=\"lazy\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-30.png\" alt=\"\" class=\"wp-image-25626\" width=\"435\" height=\"329\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-30.png 772w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-30-300x228.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2025\/03\/image-30-768x583.png 768w\" sizes=\"(max-width: 435px) 100vw, 435px\" \/><\/figure>\n\n\n\n<p><a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#%E6%94%B6%E6%95%9B%E6%80%A7%E7%A0%94%E7%A9%B6\"><\/a><\/p>\n\n\n\n<p>\u6700\u540e\uff0c\u56fe5\u663e\u793a\u4e861.3\u4ebf\u53c2\u6570GPT\u6a21\u578b\u572832K\u5e8f\u5217\u957f\u5ea6\u4e0b\uff0c\u4f7f\u7528\u5e8f\u5217\u5e76\u884c\u5ea6\u8bbe\u7f6e\u4e3a4\u7684\u60c5\u51b5\u4e0b\uff0c\u57288\u4e2aA100 GPU\u4e0a\u7684\u6536\u655b\u6027\u3002\u5bf9\u4e8eDeepSpeed\u5e8f\u5217\u5e76\u884c\uff0c\u6211\u4eec\u4f7f\u7528\u4e0d\u540c\u7684ZeRO\u9636\u6bb5\u8fdb\u884c\u4e86\u6536\u655b\u6027\u8bc4\u4f30\u3002DeepSpeed\u5e8f\u5217\u5e76\u884c\u662f\u4e00\u79cd\u7eaf\u7cfb\u7edf\u4f18\u5316\u6280\u672f\uff0c\u7528\u4e8e\u5b9e\u73b0\u957f\u5e8f\u5217Transformer\u6a21\u578b\u7684\u8bad\u7ec3\uff0c\u56e0\u6b64\u5728\u8bad\u7ec3\u6a21\u578b\u8d28\u91cf\u4e0a\u6ca1\u6709\uff08\u8d1f\u9762\uff09\u5f71\u54cd\uff0c\u5e76\u901a\u8fc7\u5b9e\u9a8c\u5f97\u5230\u4e86\u9a8c\u8bc1\uff0c\u5982\u56fe5\u6240\u793a\u3002<a target=\"_blank\" rel=\"noreferrer noopener\" href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/blob\/master\/blogs\/deepspeed-ulysses\/media\/convg.png\"><\/a><\/p>\n\n\n\n<p><em>\u56fe5\uff1a\u4f7f\u7528\u4e0d\u540cZeRO\u5185\u5b58\u4f18\u5316\u9636\u6bb5\u7684DeepSpeed\u5e8f\u5217\u5e76\u884c\u7684\u6536\u655b\u6027\u8bc4\u4f30\u3002<\/em><\/p>\n\n\n\n<h2>DeepSpeed-Ulysses\u8f6f\u4ef6\u53ef\u7528\u6027<\/h2>\n\n\n\n<p><a href=\"https:\/\/github.com\/deepspeedai\/DeepSpeed\/tree\/master\/blogs\/deepspeed-ulysses\/chinese#deepspeed-ulysses%E8%BD%AF%E4%BB%B6%E5%8F%AF%E7%94%A8%E6%80%A7\"><\/a><\/p>\n\n\n\n<p>DeepSpeed-Ulysses\u53ea\u9700\u8fdb\u884c\u5c11\u91cf\u7b80\u5355\u4ee3\u7801\u66f4\u6539\u6765\u96c6\u6210\u5230\u60a8\u7684\u4ee3\u7801\u4e2d\u3002\u4e0b\u9762\u662f\u4e00\u4e2a\u542f\u7528\u5b83\u7684\u793a\u4f8b\uff1a<\/p>\n\n\n\n<pre class=\"wp-block-preformatted\">from deepspeed.sequence.layer import DistributedAttention\n\n# \u5c06\u539f\u59cb\u7684\u81ea\u6ce8\u610f\uff08attn\uff09\u66ff\u6362\u4e3aDeepSpeed-Ulysses\u7684\u81ea\u6ce8\u610f\n\ndist_attn = DistributedAttention(attn, get_sequence_parallel_group())<\/pre>\n\n\n\n<p>\u4e0e\u5176\u4ed6\u652f\u6301\u5e8f\u5217\u5e76\u884c\u7684\u5e93\uff08\u5982Megatron-LM\uff09\u76f8\u6bd4\uff0cDeepSpeed-Ulysses\u4e0d\u9700\u8981\u8fdb\u884c\u6a21\u578b\u91cd\u6784\u3002 DeepSpeed-Ulysses\u5df2\u7ecf\u5b8c\u5168\u4e0eMegatron-DeepSpeed\u4ee3\u7801\u5e93\u96c6\u6210\u5e76\u7ecf\u8fc7\u6d4b\u8bd5\u3002\u8fd9\u610f\u5473\u7740\u5982\u679c\u60a8\u5df2\u7ecf \u5728\u4f7f\u7528\u8fd9\u4e2a\u4ee3\u7801\u5e93\u6765\u8bad\u7ec3\u5927\u578b\u8bed\u8a00\u6a21\u578b\uff0c\u60a8\u53ef\u4ee5\u65e0\u7f1d\u5730\u4f7f\u7528DeepSpeed-Ulysses\u8bad\u7ec3\u5177\u6709\u6781\u957f\u5e8f\u5217\u7684\u6a21\u578b\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u535a\u5ba2\uff1ahttps:\/\/github.com\/deepspeedai\/DeepSpeed\/blob\/master &hellip; <a href=\"http:\/\/139.9.1.231\/index.php\/2025\/03\/19\/deepspeed-ulysses\/\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">DeepSpeed Ulysses: \u8bad\u7ec3\u6781\u957f\u5e8f\u5217Transformer\u6a21\u578b\u7684\u7cfb\u7edf\u4f18\u5316<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[11,21,4,9],"tags":[],"_links":{"self":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/25559"}],"collection":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/comments?post=25559"}],"version-history":[{"count":14,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/25559\/revisions"}],"predecessor-version":[{"id":25630,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/25559\/revisions\/25630"}],"wp:attachment":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/media?parent=25559"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/categories?post=25559"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/tags?post=25559"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}