{"id":16223,"date":"2023-11-03T09:28:07","date_gmt":"2023-11-03T01:28:07","guid":{"rendered":"https:\/\/aif.amtbbs.org\/?p=16223"},"modified":"2023-11-03T09:28:07","modified_gmt":"2023-11-03T01:28:07","slug":"%e5%a4%a7%e6%a8%a1%e5%9e%8b-%e6%a8%a1%e5%9e%8b%e5%a4%a7%e4%ba%86%e9%9a%be%e5%9c%a8%e5%93%aa%e9%87%8c%ef%bc%9f","status":"publish","type":"post","link":"https:\/\/aif.amtbbs.org\/index.php\/2023\/11\/03\/16223\/","title":{"rendered":"\u5927\u6a21\u578b: \u6a21\u578b\u5927\u4e86\u96be\u5728\u54ea\u91cc\uff1f"},"content":{"rendered":"<div class=\"article-desc\">\u53ef\u4ee5\u8bf4\uff0c\u8bad\u7ec3\u5927\u6a21\u578b\uff0c\u4e0d\u4ec5\u9700\u8981\u5145\u8db3\u7684\u8ba1\u7b97\u8d44\u6e90\u548c\u6570\u636e\uff0c\u8fd8\u9700\u8981\u6df1\u539a\u7ecf\u9a8c\u548c\u6280\u80fd\uff0c\u8fd8\u9700\u8981\u4e00\u5b9a\u7684\u8010\u5fc3\u548c\u5b9a\u529b\uff0c\u5c31\u50cf\u201c\u70bc\u4e39\u201d\u4e00\u6837\u3002<\/div>\n<div id=\"postspictures\" class=\"article-content\">\n<div id=\"container\" class=\"container am-engine\" data-v-0719ea8d=\"\" data-element=\"root\">\n<p>\u5927\u5bb6\u597d\uff0c\u6211\u662fTim\u3002<\/p>\n<p>\u81ea\u4eceGPT\u6a21\u578b\u8bde\u751f\u4ee5\u6765\uff0c\u5176\u53c2\u6570\u89c4\u6a21\u5c31\u5728\u4e0d\u505c\u7684\u6269\u5927\u3002\u4f46\u6a21\u578b\u5e76\u975e\u7b80\u5355\u7684\u76f4\u63a5\u53d8\u5927\uff0c\u9700\u8981\u5728\u6570\u636e\u3001\u8c03\u5ea6\u3001\u5e76\u884c\u8ba1\u7b97\u3001\u7b97\u6cd5\u548c\u673a\u5668\u8d44\u6e90\u4e0a\u505a\u76f8\u5e94\u7684\u6539\u53d8\u3002<\/p>\n<p>\u4eca\u5929\u5c31\u6765\u603b\u7ed3\u4e0b\uff0c\u4ec0\u4e48\u662f\u5927\u6a21\u578b\uff0c\u6a21\u578b\u53d8\u5927\u7684\u96be\u5728\u54ea\u91cc\u4ee5\u53ca\u5bf9\u4e8eCV\/NLP\u6216\u8005\u641c\u63a8\u5e7f\u573a\u666f\u4e0a\u6709\u4ec0\u4e48\u5e94\u5bf9\u7b56\u7565\u3002<\/p>\n<h2>\u4ec0\u4e48\u662f\u5927\u6a21\u578b\uff1f<\/h2>\n<p>\u5927\u6a21\u578b\uff0c\u987e\u540d\u601d\u4e49\u4e3b\u6253\u7684\u5c31\u662f\u201c\u5927\u201d\u3002\u4e3b\u8981\u5305\u62ec\u51e0\u4e2a\u65b9\u9762\uff1a<\/p>\n<ol class=\"list-paddingleft-1\" data-id=\"odd3d8fc-aQLZ72QQ\">\n<li data-id=\"ld70c578-0eDePJT3\">\u6570\u636e\u89c4\u6a21\u5927\uff0c\u901a\u8fc7\u5927\u91cf\u7684\u6570\u636e\u63d0\u9ad8\u6a21\u578b\u7684\u6cdb\u5316\u80fd\u529b\u548c\u6027\u80fd\u3002<\/li>\n<li data-id=\"ld70c578-3ALiC7UR\">\u5927\u89c4\u6a21\u5e76\u884c\u8ba1\u7b97\u80fd\u529b\uff0c\u968f\u7740\u8ba1\u7b97\u786c\u4ef6\u7684\u4e0d\u65ad\u8fdb\u6b65\uff0c\u5982GPU\u548cTPU\u7684\u666e\u53ca\uff0c\u5927\u89c4\u6a21\u5e76\u884c\u8ba1\u7b97\u80fd\u529b\u7684\u63d0\u5347\u4f7f\u5f97\u8bad\u7ec3\u548c\u63a8\u7406\u5927\u6a21\u578b\u6210\u4e3a\u53ef\u80fd\u3002<\/li>\n<li data-id=\"ld70c578-M7KoiYio\">\u66f4\u201c\u5927\u201d\u6a21\u578b\u590d\u6742\u6027\uff1a\u5927\u6a21\u578b\u5177\u5907\u66f4\u6df1\u5c42\u6b21\u3001\u66f4\u590d\u6742\u7684\u7f51\u7edc\u7ed3\u6784\uff0c\u53ef\u4ee5\u6355\u6349\u66f4\u4e30\u5bcc\u7684\u7279\u5f81\u548c\u5173\u7cfb\uff0c\u63d0\u9ad8\u4e86\u6a21\u578b\u7684\u8868\u8fbe\u80fd\u529b\u3002<\/li>\n<\/ol>\n<p>\u5927\u6a21\u578b\u4e3b\u8981\u5206\u4e3a\u4e24\u7c7b\uff1a\u4e00\u4e2a\u662f\u7a00\u758f\u5927\u6a21\u578b\uff0c\u53e6\u4e00\u4e2a\u662f\u7a20\u5bc6\u5927\u6a21\u578b\u3002<\/p>\n<ol class=\"list-paddingleft-1\" data-id=\"odd3d8fc-cdbhaRik\">\n<li data-id=\"ld70c578-hOU21U7L\">\u7a00\u758f\u5927\u6a21\u578b\uff1a\u7a00\u758f\u5927\u6a21\u578b\u662f\u6307\u6a21\u578b\u4e2d\u5b58\u5728\u5927\u91cf\u7a00\u758f\u53c2\u6570\u7684\u60c5\u51b5\uff0c\u4e00\u822c\u662f\u641c\u7d22\u3001\u63a8\u8350\u3001\u5e7f\u544a\u7c7b\u4efb\u52a1\u3002\u5b83\u7684\u7279\u70b9\u662f\u6d77\u91cf\u6837\u672c\u53ca\u5927\u89c4\u6a21\u7a00\u758f\u53c2\u6570\uff08sparse embeddings\uff09\uff0c\u9002\u5408\u4f7f\u7528 CPU\/GPU \u53c2\u6570\u670d\u52a1\u5668\u6a21\u5f0f\uff08PS\uff09\u8fdb\u884c\u8bad\u7ec3\u3002<\/li>\n<li data-id=\"ld70c578-EfGFjgP5\">\u7a20\u5bc6\u5927\u6a21\u578b\uff1a\u7a20\u5bc6\u5927\u6a21\u578b\u662f\u6307\u6a21\u578b\u4e2d\u7684\u53c2\u6570\u5927\u591a\u6570\u90fd\u662f\u975e\u96f6\u503c\uff0c\u6ca1\u6709\u660e\u663e\u7684\u7a00\u758f\u6027\u7279\u5f81\uff0c\u4e00\u822c\u662fCV\u3001NLP \u4efb\u52a1\u3002\u5b83\u7684\u7279\u70b9\u662f\u5e38\u89c4\u6837\u672c\u6570\u636e\u53ca\u5927\u89c4\u6a21\u7a20\u5bc6\u53c2\u6570\uff0c\u5b83\u9002\u5408\u7528\u7eaf GPU \u96c6\u5408\u901a\u4fe1\u6a21\u5f0f\uff08Collective\uff09\u8fdb\u884c\u8bad\u7ec3\u3002<\/li>\n<\/ol>\n<p>\u5bf9\u4e8e\u641c\u63a8\u5e7f\u7c7b\u7684\u7a00\u758f\u5927\u6a21\u578b\u6765\u8bf4\uff0c\u4e00\u822c\u5305\u542b\u7a00\u758f\u7279\u5f81\u7684\u5d4c\u5165\uff08embedding\uff09\u548c\u7a20\u5bc6\u6a21\u578b\u4e24\u4e2a\u90e8\u5206\u3002<\/p>\n<p>\u5176\u4e2d\uff0c\u7a00\u758f\u7279\u5f81\u7684\u5d4c\u5165\u8ba1\u7b97\u662f\u7a00\u758f\u5927\u6a21\u578b\u7684\u5173\u952e\uff0c\u800c\u7a20\u5bc6\u6a21\u578b\u90e8\u5206\u4e00\u822c\u5f80\u5f80\u8f83\u5c0f\uff0c\u53ef\u4ee5\u653e\u5230\u4e00\u4e2aGPU\u5185\uff0c\u56e0\u6b64\u53ef\u4ee5\u8fdb\u884cdata\u5e76\u884c\u4ee5\u53caall reduce\u901a\u8baf\u3002<\/p>\n<p>\u5728\u8bad\u7ec3\u4e2d\uff0c\u9700\u8981\u5728\u7279\u5f81\u5d4c\u5165\u8868\uff08embedding table\uff09\u4e0a\u9700\u8981\u8fdb\u884c\u590d\u6742\u7684\u67e5\u627e\u3001\u6392\u5217\u7b49\u64cd\u4f5c\uff0c\u7136\u540e\u751f\u6210\u5f20\u91cf\u518d\u505a\u7a20\u5bc6\u6a21\u578b\u7684\u8ba1\u7b97\u3002<\/p>\n<p>\u800c\u7279\u5f81\u5d4c\u5165\u8868\u5f80\u5f80\u4f1a\u5360\u7528\u975e\u5e38\u5927\u7684\u5b58\u50a8\u7a7a\u95f4\uff0c\u9700\u8981\u5f88\u591a\u53f0GPU\u670d\u52a1\u5668\u624d\u80fd\u5b8c\u6574\u5b58\u653e\uff0c\u8fd9\u5c31\u662f\u5178\u578b\u7684tensor\u5e76\u884c\u3002<\/p>\n<p>\u5728\u8fd9\u6837\u7684\u573a\u666f\u4e0b\uff0c\u5c31\u4f1a\u5bfc\u81f4\u5178\u578b\u7684alltoall\u7684\u901a\u8baf\u6a21\u5f0f\uff0c\u800calltoall\u901a\u8baf\u4f1a\u5e26\u6765\u4e25\u91cd\u7684incast\u901a\u8baf\uff08\u591a\u6253\u4e00\uff09\uff0c\u8fdb\u800c\u5e26\u6765\u7f51\u7edc\u62e5\u585e\uff0c\u7ed9\u7f51\u7edc\u67b6\u6784\u3001\u62e5\u585e\u63a7\u5236\u534f\u8bae\u3001\u8d1f\u8f7d\u5747\u8861\u7b97\u6cd5\u7b49\u90fd\u63d0\u51fa\u4e86\u5f88\u9ad8\u7684\u8981\u6c42\u3002<\/p>\n<p>\u5bf9\u4e8eCV\u3001NLP \u4efb\u52a1\u6765\u8bf4\uff0c\u7531\u4e8e\u6a21\u578b\u53c2\u6570\u975e\u5e38\u591a\uff0c\u8fdc\u8fdc\u8d85\u8fc7\u4e86\u5355\u4e2aGPU\u663e\u5b58\u6240\u80fd\u5bb9\u7eb3\u7684\u7a7a\u95f4\uff08NVIDIA\u6700\u65b0\u7684A100\u4e5f\u5c31\u662f80GB\u663e\u5b58\uff09\u3002<\/p>\n<p>\u6240\u4ee5\u5f80\u5f80\u65e2\u9700\u8981\u5bf9\u6a21\u578b\u67d0\u4e00\u5c42\u7684tensor\u5e76\u884c\uff0c\u4e5f\u9700\u8981\u4e0d\u540c\u5c42\u4e4b\u95f4\u7684pipeline\u5e76\u884c\uff0c\u624d\u80fd\u653e\u5f97\u4e0b\u6574\u4e2a\u5927\u6a21\u578b\u3002<\/p>\n<p>\u5728\u8ba1\u7b97\u8fc7\u7a0b\u4e2d\uff0c\u65e2\u6709\u5355\u673a\u5185\u901a\u8baf\uff0c\u4e5f\u6709\u4e0d\u540c\u673a\u5668\u95f4\u7684\u901a\u8baf\uff0c\u5177\u4f53\u7684\u901a\u8baf\u6a21\u5f0f\u53d6\u51b3\u4e8e\u6a21\u578b\u7684\u5207\u5206\u548c\u653e\u7f6e\u65b9\u6cd5\u3002<\/p>\n<p>\u4e3a\u4e86\u52a0\u901f\u8bad\u7ec3\u8fc7\u7a0b\uff0c\u5f80\u5f80\u5b8c\u6574\u7684\u5927\u6a21\u578b\u4e4b\u95f4\u4e5f\u4f1a\u91c7\u7528data\u5e76\u884c\uff0c\u6bcf\u4e00\u4e2a\u5b8c\u6574\u7684\u5927\u6a21\u578b\u4f1a\u88ab\u6295\u5582\u4e0d\u540c\u7684\u8bad\u7ec3\u6570\u636e\uff0c\u8fd9\u5c31\u4f1a\u5bfc\u81f4\u5927\u5bb6\u719f\u6089\u7684allreduce\u901a\u4fe1\u6a21\u5f0f\u3002<\/p>\n<p>\u603b\u7ed3\u8d77\u6765\uff0c\u7a20\u5bc6\u5927\u6a21\u578b\u548c\u7a00\u758f\u5927\u6a21\u578b\u5728\u6a21\u578b\u7279\u5f81\u4e0a\u6709\u7740\u660e\u663e\u7684\u5dee\u5f02\uff0c\u5bf9\u8ba1\u7b97\/\u5b58\u50a8\/\u901a\u4fe1\u8d44\u6e90\u7684\u9700\u6c42\u4e5f\u5b58\u5728\u660e\u663e\u7684\u4e0d\u540c\u3002<\/p>\n<p>\u8981\u8fbe\u5230GPU\u7b97\u529b\u8d44\u6e90\u7684\u6700\u5927\u5316\u5229\u7528\u548c\u6700\u597d\u7684\u52a0\u901f\u6548\u679c\uff0c\u9700\u8981\u7ed3\u5408\u6a21\u578b\u7279\u5f81\u548c\u5b9e\u73b0\u65b9\u5f0f\u5bf9GPU\u670d\u52a1\u5668\u67b6\u6784\u3001\u7f51\u7edc\u67b6\u6784\u3001\u8bad\u7ec3\u6570\u636e\u5b58\u50a8\u548c\u62c9\u53d6\u3001\u5206\u5e03\u5f0f\u8bad\u7ec3\u6846\u67b6\u8fdb\u884c\u5168\u5c40\u7684\u8003\u91cf\u548c\u8bbe\u8ba1\u3002<\/p>\n<h2>\u6a21\u578b\u5927\u4e86\u96be\u5728\u54ea\u91cc\uff1f<\/h2>\n<p>\u5927\u6a21\u578b\u5e26\u6765\u7684\u6311\u6218\u4e3b\u8981\u6709\u4e24\u70b9\uff1a\u6d77\u91cf\u6837\u672c\u3001\u53c2\u6570\uff08\u4e07\u4ebf\u7ea7\u522b\uff09\u548c\u8f83\u957f\u7684\u6536\u655b\u65f6\u95f4\u3002<\/p>\n<p>\u522b\u770b\u53ea\u6709\u8fd9\u533a\u533a\u4e24\u70b9\uff0c\u5b83\u4f1a\u884d\u751f\u51fa\u5f88\u591a\u8981\u89e3\u51b3\u7684\u95ee\u9898\u3002<\/p>\n<h3>1. \u5927\u6a21\u578b\u8bad\u7ec3\u9700\u8981\u66f4\u5927\u7684\u7b97\u529b\u3002<\/h3>\n<p>\u5927\u6a21\u578b\u8bad\u7ec3\u6240\u9700\u7684\u603b\u7b97\u529b\uff0c\u5176\u5b9e\u5f88\u7b80\u5355\uff0c6 * \u6a21\u578b\u7684\u53c2\u6570\u91cf * \u8bad\u7ec3\u6570\u636e\u7684 token \u6570\u5c31\u662f\u6240\u6709\u8bad\u7ec3\u6570\u636e\u8fc7\u4e00\u904d\u6240\u9700\u7684\u7b97\u529b\u3002\u8fd9\u91cc\u7684 6 \u5c31\u662f\u6bcf\u4e2a token \u5728\u6a21\u578b\u6b63\u5411\u4f20\u64ad\u548c\u53cd\u5411\u4f20\u64ad\u7684\u65f6\u5019\u6240\u9700\u7684\u4e58\u6cd5\u3001\u52a0\u6cd5\u8ba1\u7b97\u6b21\u6570\u3002<\/p>\n<p>\u4e00\u5806\u77e9\u9635\u76f8\u4e58\uff0c\u7b80\u5355\u6765\u60f3\u5c31\u662f\u5de6\u8fb9\u82e5\u5e72\u4e2a\u795e\u7ecf\u5143\uff0c\u53f3\u8fb9\u82e5\u5e72\u4e2a\u795e\u7ecf\u5143\uff0c\u7ec4\u6210\u4e00\u4e2a\u5b8c\u5168\u4e8c\u5206\u56fe\u3002\u9009\u51fa\u5176\u4e2d\u4efb\u610f\u4e00\u4e2a\u5de6\u8fb9\u7684\u795e\u7ecf\u5143 L \u548c\u53f3\u8fb9\u7684\u795e\u7ecf\u5143 R\u3002<\/p>\n<ul class=\"list-paddingleft-1\" data-id=\"u4533b8e-4gdWkXfH\">\n<li data-id=\"ld70c578-3U1mAc1W\">\u6b63\u5411\u4f20\u64ad\u7684\u65f6\u5019\uff1a L\u628a\u5b83\u7684\u8f93\u51fa\u4e58\u4e0a L\u548c R \u4e4b\u95f4\u7684\u6743\u91cd w\uff0c\u53d1\u7ed9 R\uff1bR\u4e0d\u53ef\u80fd\u53ea\u8fde\u4e00\u4e2a\u795e\u7ecf\u5143\u5427\uff0c\u603b\u8981\u628a\u591a\u4e2a L\u7684\u52a0\u5230\u4e00\u8d77\uff0c\u8fd9\u5c31\u662f reduce\uff0c\u9700\u8981\u4e00\u6b21\u52a0\u6cd5\u3002<\/li>\n<li data-id=\"ld70c578-5CNpE5WD\">R\u628a\u5b83\u6536\u5230\u7684\u68af\u5ea6\u4e58\u4e0a L\u548c R \u4e4b\u95f4\u7684\u6743\u91cd w\uff0c\u53d1\u7ed9 L\uff1bL\u4e5f\u4e0d\u53ef\u80fd\u53ea\u8fde\u4e00\u4e2a R\uff0c\u9700\u8981\u628a\u68af\u5ea6 reduce \u4e00\u4e0b\uff0c\u505a\u4e2a\u52a0\u6cd5\uff1b\u522b\u5fd8\u4e86\u6743\u91cd w \u9700\u8981\u66f4\u65b0\uff0c\u90a3\u5c31\u8981\u8ba1\u7b97 w \u7684\u68af\u5ea6\uff0c\u628a R \u6536\u5230\u7684\u68af\u5ea6\u4e58\u4e0a L\u6b63\u5411\u4f20\u64ad\u7684\u8f93\u51fa\uff08activation\uff09\uff1b\u4e00\u4e2a\u00a0batch\u00a0\u4e00\u822c\u6709\u591a\u4e2a sample\uff0c\u6743\u91cd w \u7684\u66f4\u65b0\u9700\u8981\u628a\u8fd9\u4e9b sample \u7684\u68af\u5ea6\u52a0\u5230\u4e00\u8d77\u3002<\/li>\n<li data-id=\"ld70c578-lmQht1qM\">\u4e00\u5171 3 \u6b21\u4e58\u6cd5\uff0c3 \u6b21\u52a0\u6cd5\uff0c\u4e0d\u7ba1 Transformer \u591a\u590d\u6742\uff0c\u77e9\u9635\u8ba1\u7b97\u5c31\u662f\u8fd9\u4e48\u7b80\u5355\uff0c\u5176\u4ed6\u7684\u5411\u91cf\u8ba1\u7b97\u3001softmax\u00a0\u4e4b\u7c7b\u7684\u90fd\u4e0d\u662f\u5360\u7b97\u529b\u7684\u4e3b\u8981\u56e0\u7d20\uff0c\u4f30\u7b97\u7684\u65f6\u5019\u53ef\u4ee5\u5ffd\u7565\u3002<\/li>\n<\/ul>\n<p>\u6709\u4e86\u6a21\u578b\u8bad\u7ec3\u6240\u9700\u7684\u603b\u7b97\u529b\uff0c\u9664\u4ee5\u6bcf\u4e2a GPU \u7684\u7406\u8bba\u7b97\u529b\uff0c\u518d\u9664\u4ee5 GPU \u7684\u6709\u6548\u7b97\u529b\u5229\u7528\u6bd4\u4f8b\uff0c\u5c31\u5f97\u5230\u4e86\u6240\u9700\u7684 GPU-hours\u3002<\/p>\n<p>\u5982\u679c\u8bad\u7ec3\u4e00\u4e2a\u901a\u7528\u5927\u8bed\u8a00\u6a21\u578b\u7684\u57fa\u5ea7\u9700\u8981\u534a\u5e74\u6216\u51e0\u4e2a\u6708\u7684\u65f6\u95f4\uff0c\u540c\u65f6\u8fd8\u5360\u7528\u975e\u5e38\u591a\u7684\u673a\u5668\u8d44\u6e90\uff0c\u8fd9\u5c31\u4f7f\u5f97\u5927\u6a21\u578b\u7684\u8bad\u7ec3\u975e\u5e38\u201c\u8d35\u201d\uff0c\u4f7f\u5f97\u5927\u6a21\u578b\u7684\u8bad\u7ec3\u6210\u4e3a\u4e86\u4e2a\u522b\u5927\u4f01\u4e1a\u7684\u4e13\u7528\u3002<\/p>\n<p>\u4f8b\u5982\uff0c\u5982\u679c\u4f60\u6709100\u4e2a\u5b9e\u9a8c\u60f3\u8bd5\u8bd5\uff0c\u800c\u6a21\u578b\u8bad\u7ec3\u9700\u8981\u534a\u5e74\uff0c\u90a3\u4f60\u53ea\u80fd\u5728\u5176\u4e2d\u9009\u62e9\u4f18\u5148\u7ea7\u9ad8\u8fdb\u884c\u5b9e\u9a8c\u3002<\/p>\n<p>\u8fd9\u4e5f\u5c31\u662f\u4e3a\u4ec0\u4e48\u73b0\u5728\u5927\u5bb6\u7684\u5927\u8bed\u8a00\u6a21\u578b\u57fa\u5ea7\u90fd\u662f\u8ffd\u8e2a\u6700\u65b0\u5f00\u6e90\u7684\u6a21\u578b\u3002\u56e0\u4e3a\u81ea\u5df1\u641e\u901a\u7528\u5927\u6a21\u578b\u57fa\u5ea7\uff0c\u4e00\u65b9\u9762\u641e\u534a\u5929\u6210\u672c\u4e0a\u8017\u4e0d\u8d77\uff0c\u53e6\u4e00\u65b9\u9762\u53ef\u80fd\u8fd8\u6ca1\u5565\u6548\u679c\u3002<\/p>\n<h3>2. \u5927\u6a21\u578b\u8bad\u7ec3\u9700\u8981\u66f4\u591a\u7684\u663e\u5b58\u5185\u5b58\u8d44\u6e90\u3002<\/h3>\n<p>\u6df1\u5ea6\u5b66\u4e60\u8bad\u7ec3\u9700\u8981\u7684\u5185\u5b58\u5305\u62ec\u6a21\u578b\u53c2\u6570\u3001\u53cd\u5411\u4f20\u64ad\u7684\u68af\u5ea6\u3001\u4f18\u5316\u5668\u6240\u7528\u7684\u5185\u5b58\u3001\u6b63\u5411\u4f20\u64ad\u7684\u4e2d\u95f4\u72b6\u6001\uff08activation\uff09\uff0c\u663e\u5b58\u5360\u7528 = \u6a21\u578b\u53c2\u6570\u5927\u5c0f + Batch Size * \u4f18\u5316\u5668\u53c2\u6570\u4e0e\u4e2d\u95f4\u53d8\u91cf\u7684\u5927\u5c0f\u3002<\/p>\n<ul class=\"list-paddingleft-1\" data-id=\"u4533b8e-VGW0cWpC\">\n<li data-id=\"ld70c578-Bimch7BW\">\u4f18\u5316\u5668\u6240\u7528\u7684\u5185\u5b58\u7684\u8ba1\u7b97\u5176\u5b9e\u4e5f\u5f88\u7b80\u5355\uff0c\u5982\u679c\u7528\u6700\u7ecf\u5178\u7684 Adam \u4f18\u5316\u5668\uff0c\u5b83\u9700\u8981\u7528 32 \u4f4d\u6d6e\u70b9\u6765\u8ba1\u7b97\u3002\u5373\u4f7f\u6211\u4eec\u4f7f\u7528mixed-precision\u8fdb\u884c\u8ba1\u7b97\uff0c\u6bcf\u4e2a\u53c2\u6570\u9700\u8981\u4e5f\u8981\u5b58 4 \u5b57\u8282\u7684 32 \u4f4d\u7248\u672c\uff08\u6b63\u5411\u4f20\u64ad\u65f6\u7528 16 \u4f4d\u7248\u672c\uff0c\u4f18\u5316\u65f6\u7528 32 \u4f4d\u7248\u672c\uff09\uff0c\u8fd8\u9700\u8981\u5b58 4 \u5b57\u8282\u7684 momentum \u548c 4 \u5b57\u8282\u7684 variance\uff0c\u4e00\u5171 12 \u5b57\u8282\u3002\u5982\u679c\u662f\u7528\u7c7b\u4f3c SGD \u7684\u4f18\u5316\u5668\uff0c\u53ef\u4ee5\u4e0d\u5b58 variance\uff0c\u53ea\u9700\u8981 8 \u5b57\u8282\u3002<\/li>\n<li data-id=\"ld70c578-t5QEaD93\">\u6b63\u5411\u4f20\u64ad\u7684\u4e2d\u95f4\u72b6\u6001\uff08activation\uff09\u662f\u53cd\u5411\u4f20\u64ad\u65f6\u8ba1\u7b97\u68af\u5ea6\u5fc5\u9700\u7684\uff0c\u800c\u4e14\u8ddf batch size \u6210\u6b63\u6bd4\u3002Batch size \u8d8a\u5927\uff0c\u6bcf\u6b21\u8bfb\u53d6\u6a21\u578b\u53c2\u6570\u5185\u5b58\u80fd\u505a\u7684\u8ba1\u7b97\u5c31\u8d8a\u591a\uff0c\u8fd9\u6837\u5bf9 GPU \u5185\u5b58\u5e26\u5bbd\u7684\u538b\u529b\u5c31\u8d8a\u5c0f\u3002\u5212\u91cd\u70b9\uff1a\u6b63\u5411\u4f20\u64ad\u7684\u4e2d\u95f4\u72b6\u6001\u6570\u91cf\u662f\u8ddf batch size \u6210\u6b63\u6bd4\u7684\u3002<\/li>\n<\/ul>\n<p>\u5f53\u7136\u4e5f\u6709\u8282\u7701\u5185\u5b58\u8d44\u6e90\u7684\u529e\u6cd5\uff0c\u4f8b\u5982\u7b97\u529b\u6362\u5185\u5b58\uff0c\u65f6\u95f4\u6362\u5185\u5b58\u7b49\u3002<\/p>\n<p>\u7b97\u529b\u6362\u5185\u5b58\u7684\u628a\u620f\uff0c\u5c31\u662f\u4e0d\u8981\u5b58\u50a8\u90a3\u4e48\u591a\u68af\u5ea6\u548c\u6bcf\u4e00\u5c42\u7684\u6b63\u5411\u4f20\u64ad\u7684\u4e2d\u95f4\u72b6\u6001\uff0c\u800c\u662f\u5728\u8ba1\u7b97\u5230\u67d0\u4e00\u5c42\u7684\u65f6\u5019\u518d\u4e34\u65f6\u4ece\u5934\u5f00\u59cb\u91cd\u7b97\u6b63\u5411\u4f20\u64ad\u7684\u4e2d\u95f4\u72b6\u6001\u3002<\/p>\n<p>\u5982\u679c\u6bcf\u4e00\u5c42\u90fd\u8fd9\u4e48\u5e72\uff0c\u90a3\u4e48\u5c31\u53ea\u8981 2 \u4e2a\u5b57\u8282\u6765\u5b58\u8fd9\u4e00\u5c42\u7684\u68af\u5ea6\uff0c\u4f46\u662f\u8ba1\u7b97\u4e2d\u95f4\u72b6\u6001\u7684\u7b97\u529b\u5f00\u9500\u4f1a\u5f88\u5927\u3002<\/p>\n<p>\u56e0\u6b64\u5b9e\u9645\u4e2d\u4e00\u822c\u662f\u628a\u6574\u4e2a Transformer \u5206\u6210\u82e5\u5e72\u7ec4\uff0c\u4e00\u7ec4\u6709\u82e5\u5e72\u5c42\uff0c\u53ea\u4fdd\u5b58\u6bcf\u7ec4\u7b2c\u4e00\u5c42\u7684\u4e2d\u95f4\u72b6\u6001\uff0c\u540e\u9762\u7684\u5c42\u5c31\u4ece\u8be5\u7ec4\u7b2c\u4e00\u5c42\u5f00\u59cb\u91cd\u65b0\u8ba1\u7b97\uff0c\u8fd9\u6837\u5c31\u5e73\u8861\u4e86\u7b97\u529b\u548c\u5185\u5b58\u7684\u5f00\u9500\u3002<\/p>\n<p>\u65f6\u95f4\u6362\u5185\u5b58\u7684\u628a\u620f\uff0c\u6309\u987a\u5e8f\u6267\u884cMini-Batch\u6570\u636e\u7684\u524d\u5411\u8ba1\u7b97\u68af\u5ea6\uff0c\u540c\u65f6\u5bf9\u68af\u5ea6\u8fdb\u884c\u7d2f\u79ef\uff0c\u7d2f\u79ef\u7684\u7ed3\u679c\u5728\u6700\u540e\u4e00\u4e2aMini-Batch\u8ba1\u7b97\u540e\u6c42\u5e73\u5747\u66f4\u65b0\u6a21\u578b\u53d8\u91cf\u3002<\/p>\n<p>\u6b64\u5916\uff0c\u8fd8\u53ef\u4ee5\u591a\u7ea7\u7f13\u5b58\uff0cGPU \u5185\u5b58\u653e\u4e0d\u4e0b\u53ef\u4ee5\u6362\u51fa\u5230 CPU \u5185\u5b58\u3002<\/p>\n<p>\u4f8b\u5982\uff0c\u5bf9\u4e8e LLaMA-2 70B \u6a21\u578b\uff0c\u6a21\u578b\u53c2\u6570\u9700\u8981 140 GB\uff0c\u53cd\u5411\u4f20\u64ad\u7684\u68af\u5ea6\u9700\u8981 140 GB\uff0c\u4f18\u5316\u5668\u7684\u72b6\u6001\uff08\u5982\u679c\u7528 Adam\uff09\u9700\u8981 840 GB\u3002<\/p>\n<h3>3. \u5bf9\u6570\u636e\u7684\u6570\u91cf\u548c\u6570\u636e\u7684\u8d28\u91cf\u8981\u6c42\u6781\u9ad8<\/h3>\n<p>\u5bf9\u4e8e\u6d77\u91cf\u6570\u636e\u6837\u672c\u6765\u8bf4\uff0c\u5e76\u4e0d\u662f\u90fd\u5582\u8fdb\u53bb\u5c31\u6548\u679c\u597d\uff0c\u54ea\u4e9b\u6570\u636e\u6709\u4ef7\u503c\uff0c\u54ea\u4e9b\u6ca1\u4ef7\u503c\u3002\u7531\u4e8e\u6570\u636e\u91cf\u7684\u589e\u52a0\uff0c\u5206\u8fa8\u6570\u636e\u7684\u4ef7\u503c\u4e5f\u5e26\u6765\u5f88\u5927\u7684\u56f0\u96be\u3002<\/p>\n<p>\u6b64\u5916\uff0c\u5927\u91cf\u7684\u6570\u636e\u5b58\u50a8\u5728\u54ea\u91cc\uff0c\u4e00\u822c\u53ef\u4ee5\u5b58\u50a8\u5728HDFS\u6216S3\u3002\u4f46\u600e\u4e48\u4fdd\u8bc1\u5b58\u53d6\u80fd\u4e0d\u628a\u673a\u5668\u585e\u6ee1\u4e14\u80fd\u5feb\u901f\u8c03\u53d6\uff0c\u5bf9\u4e8e\u76f8\u540c\u6570\u636e\u7684\u6a21\u578b\u591a\u6b21\u8bad\u7ec3\uff0c\u662f\u5426\u53ef\u4ee5\u901a\u8fc7cache\u6765\u52a0\u901f\u6a21\u578b\u8bad\u7ec3\u7684\u65f6\u95f4\u3002<\/p>\n<h3>4. Transformer\u7684\u201c\u4e0d\u53ef\u80fd\u4e09\u89d2\u201d\u7279\u6027<\/h3>\n<p><img data-dominant-color=\"ececec\" data-has-transparency=\"false\" style=\"--dominant-color: #ececec;\" loading=\"lazy\" decoding=\"async\" class=\"not-transparent alignnone size-full wp-image-16224\" src=\"https:\/\/aiforumimage.oss-cn-shanghai.aliyuncs.com\/wp-content\/uploads\/2023\/11\/c15601112a06a15f214737433fc7d2577918d7.png\" width=\"818\" height=\"750\" alt=\"\u56fe\u7247\" title=\"\u56fe\u7247\" srcset=\"https:\/\/aiforumimage.oss-cn-shanghai.aliyuncs.com\/wp-content\/uploads\/2023\/11\/c15601112a06a15f214737433fc7d2577918d7.png 818w, https:\/\/aiforumimage.oss-cn-shanghai.aliyuncs.com\/wp-content\/uploads\/2023\/11\/c15601112a06a15f214737433fc7d2577918d7-300x275.png 300w, https:\/\/aiforumimage.oss-cn-shanghai.aliyuncs.com\/wp-content\/uploads\/2023\/11\/c15601112a06a15f214737433fc7d2577918d7-150x138.png 150w, https:\/\/aiforumimage.oss-cn-shanghai.aliyuncs.com\/wp-content\/uploads\/2023\/11\/c15601112a06a15f214737433fc7d2577918d7-768x704.png 768w\" sizes=\"auto, (max-width: 818px) 100vw, 818px\" \/><span class=\"data-image-title\">\u56fe\u7247<\/span><\/p>\n<p>Transformer\u5728\u8bad\u7ec3\u5e76\u884c\u6027\u3001\u63a8\u7406\u6548\u7387\u548c\u7ade\u4e89\u6027\u80fd\u4e4b\u95f4\u5f88\u96be\u53d6\u5f97\u5e73\u8861\u88ab\u79f0\u4e3a\u201c\u4e0d\u53ef\u80fd\u4e09\u89d2\u201d\u3002<\/p>\n<p>Transformers \u7531\u4e8e\u5176\u56fa\u6709\u7684\u6bcf\u6b65 O(N) \u590d\u6742\u5ea6\u548c\u5185\u5b58\u9650\u5236\u7684\u952e\u503c\u7f13\u5b58\uff0c\u5728\u63a8\u7406\u8fc7\u7a0b\u4e2d\u8868\u73b0\u51fa\u6b21\u4f18\u6548\u7387\u3002\u8fd9\u79cd\u4f4e\u6548\u7387\u4f7f\u5b83\u4eec\u7684\u5b9e\u9645\u90e8\u7f72\u53d8\u5f97\u590d\u6742\uff0c\u7279\u522b\u662f\u5bf9\u4e8e\u957f\u5e8f\u5217\u3002<\/p>\n<h3>5. \u6df1\u5ea6\u5b66\u4e60\u6846\u67b6\u4e0a\u4e0d\u662f\u5f88\u53cb\u597d<\/h3>\n<p>\u56e0\u4e3a\u8fd9\u4e2a\u9886\u57df\u6700\u8fd1\u51e0\u5e74\u624d\u5f00\u59cb\u70ed\u95e8\uff0c\u800c\u4e4b\u524d\u7684\u6846\u67b6pytorch\u3001tensorflow\u7b49\u662f\u65e9\u5c31\u51fa\u73b0\u7684\uff0c\u5f53\u65f6\u5e76\u6ca1\u6709\u9488\u5bf9\u5927\u6a21\u578b\u7684\u5206\u5e03\u5f0f\u8bad\u7ec3\u7684\u9700\u6c42\u573a\u666f\u505a\u6df1\u5165\u7684\u62bd\u8c61\u8bbe\u8ba1\u548c\u4f18\u5316\u3002<\/p>\n<p>\u6240\u4ee5\u8fd9\u4e2a\u9886\u57df\u9700\u8981\u901a\u8fc7\u5927\u6570\u636e\u6846\u67b6\u5230\u6df1\u5ea6\u5b66\u4e60\u6846\u67b6\u7684\u7aef\u5230\u7aef\u6253\u901a\uff0c\u5f62\u6210\u8fd9\u6837\u7684\u4e00\u5957\u65b0\u7684\u7f16\u7a0b\u8303\u5f0f\u548c\u5bf9\u5e94\u7684\u8ba1\u7b97\u6846\u67b6\u6765\u89e3\u51b3\u6389\u3002<\/p>\n<p>\u4e0b\u9762\u6211\u4eec\u4ee5CV\u3001NLP\u573a\u666f\u548c\u641c\u63a8\u5e7f\u573a\u666f\u8fdb\u884c\u5206\u522b\u8be6\u7ec6\u8bf4\u660e\u3002<\/p>\n<p>CV\u548cNLP\u573a\u666f\uff1a<\/p>\n<p>\u5bf9CV\u548cNLP\u573a\u666f\u6765\u8bf4\uff0c\u5176\u7279\u70b9\u4e3b\u8981\u6709\uff1a<\/p>\n<ol class=\"list-paddingleft-1\" data-id=\"odd3d8fc-nFkjn3M8\">\n<li data-id=\"ld70c578-A7N6MvFw\">\u6a21\u578b\u4e00\u822c\u590d\u6742\uff0c\u5355\u673a\u6027\u80fd\u8981\u6c42\u9ad8\u3002\u4e1a\u754c\u4e3b\u8981\u4f7f\u7528\u9ad8\u6027\u80fd\u7684GPU\u8fdb\u884c\u8ba1\u7b97\uff0c\u5e76\u91c7\u7528All-reduce\u7684\u901a\u4fe1\u62d3\u6251\u8fdb\u884c\u53c2\u6570\u7684\u540c\u6b65\u66f4\u65b0\u3002<\/li>\n<li data-id=\"ld70c578-G5bGJ5TP\">\u6a21\u578b\u5927\uff08DenseNet\u90e8\u5206\uff09\uff0c\u6bd4\u5982NLP\u9886\u57df\uff0cGPT-3\u8fd9\u6837\u7684\u6a21\u578b\u9ad8\u8fbe1750\u4ebf\u53c2\u6570\uff0c\u663e\u5b58\u5360\u7528\u9ad8\u8fbe2.8 TB\uff0c\u5355\u673a\u5185\u5b58\u65e0\u6cd5\u5bb9\u7eb3\u3002<\/li>\n<\/ol>\n<p>\u5f53\u9762\u5bf9GPT-3\u8fd9\u79cdDense\u90e8\u5206\u5927\u7684\u6a21\u578b\uff0cAllreduce \u5355\u5361\u5185\u5b58\u65e0\u6cd5\u5bb9\u7eb3\uff0c\u6211\u4eec\u9700\u8981\u91c7\u7528\u6a21\u578b\u5e76\u884c(model parallelism)\u7684\u65b9\u5f0f\u5c06\u8ba1\u7b97\u56fe\u5212\u5206\u5230\u4e0d\u540c\u7684\u8bbe\u5907\u4e0a\u6784\u5efa\u6709\u5411\u65e0\u73af\u56fe(DAG)\u8fdb\u884c\u5206\u5e03\u5f0f\u8bad\u7ec3\uff0c\u5176\u4e2dGpipe, Megatron,\u00a0Oneflow\u548cWhale\u90fd\u63d0\u51fa\u6a21\u578b\u5e76\u884c\u7684\u76f8\u5173\u89e3\u51b3\u65b9\u6848\u3002<\/p>\n<p>\u76f8\u6bd4\u4e8e\u6570\u636e\u5e76\u884c\u6bcf\u4e2aworker\u53ea\u6709\u4e00\u90e8\u5206\u6570\u636e\uff0c\u6a21\u578b\u5e76\u884c\u4e0b\u6bcf\u4e2anode\u4f7f\u7528\u6240\u6709\u6570\u636e\u3002<\/p>\n<p>\u4e0b\u9762\u6211\u4eec\u7b80\u5355\u8bf4\u660e\u51e0\u79cd\u6a21\u578b\u5e76\u884c\u7684\u65b9\u6cd5\uff1a<\/p>\n<ol class=\"list-paddingleft-1\" data-id=\"odd3d8fc-4oTSJTSZ\">\n<li data-id=\"ld70c578-GKY1iY4e\">Tensor Parallelism\uff0c\u4e3b\u8981\u662f\u5c06\u4e00\u5c42Layer\u4e2d\u7684\u77e9\u9635\u8ba1\u7b97\u5206\u522b\u62c6\u5206\u5230\u4e0d\u540c\u7684\u673a\u5668\u4e0a\u8fdb\u884c\u8fd0\u7b97\uff0c\u6bd4\u59821D Megatron\u3002<\/li>\n<li data-id=\"ld70c578-jRJimrpu\">Pipeline Parallelism\uff0c\u4f1a\u5c06\u6a21\u578b\u7684layers\u62c6\u5206\u5230\u4e0d\u540c\u7684\u673a\u5668\u4e0a\uff0c\u5219\u4e00\u6b21forward\u5c31\u9700\u8981\u8de8\u8fc7\u4e0d\u540c\u7684\u673a\u5668\u4e32\u884c\u5730\u8fdb\u884c\u8fd0\u7b97\uff0c\u800c\u6d41\u884c\u5e76\u884c\u901a\u8fc7\u5c06batch size\u5207\u5206\u4e3a\u66f4\u5c0f\u7684mirco batch\uff0c\u51cf\u5c11\u6570\u636e\u4f9d\u8d56\uff0c\u4ece\u800c\u5c06\u6574\u4e2a\u8ba1\u7b97\u8fc7\u7a0b\u5f02\u6b65\u8d77\u6765\uff0c\u6700\u5927\u5316\u8d44\u6e90\u5229\u7528\u7387\u3002<\/li>\n<\/ol>\n<h3>CTR\u63a8\u5e7f\u641c\u573a\u666f\uff1a<\/h3>\n<p>\u5bf9\u4e8eCTR\u5927\u6a21\u578b\u573a\u666f\u6765\u8bf4\uff0c\u5176\u5177\u6709\u6a21\u578b\u5c0f\uff0c\u8bcd\u8868\u5927\u7684\u7279\u70b9\u3002<\/p>\n<ol class=\"list-paddingleft-1\" data-id=\"odd3d8fc-l5npcL9Y\">\n<li data-id=\"ld70c578-vrhicX4q\">\u6a21\u578b\u4e2d\u7684Dense\u90e8\u5206\uff0c\u4e00\u822c\u5f88\u5c0f\uff0c\u5f80\u5f80\u4e00\u53f0\u673a\u5668\u7684\u5185\u5b58\u5c31\u53ef\u4ee5\u5bb9\u7eb3\u3002\u4f46\u662f\u5176\u7279\u5f81\u91cf\u7ea7\u53ef\u80fd\u9ad8\u8fbe\u6210\u767e\u4e0a\u5343\u4ebf\uff0c\u9020\u6210Sparse\u90e8\u5206\u6216\u8005Embedding table\u9ad8\u8fbeTB\u7ea7\u522b\uff0c\u4f7f\u5f97\u5355\u673a\u65e0\u6cd5\u5bb9\u7eb3\u3002<\/li>\n<li data-id=\"ld70c578-97s0iZgE\">\u4e00\u4e2aBatch\u7684embedding lookup\u91cf\u7ea7\u5927\uff0c\u9020\u6210\u67e5\u8be2\u8017\u65f6\u5927\u3002\u7531\u4e8e\u7279\u5f81\u6570\u91cf\u591a\uff0c\u4e00\u4e2aBatch\u53ef\u80fd\u5305\u542b\u51e0\u5341\u4e07\u4e2aID\u7c7b\u7279\u5f81\uff0cTF\u539f\u751f\u7684embedding lookup\u67e5\u8be2\u8017\u65f6\u5927\uff0c\u9020\u6210\u8bad\u7ec3\u548cinference\u6027\u80fd\u4f4e\u3002<\/li>\n<li data-id=\"ld70c578-dyc5jnrZ\">\u6570\u636e\u5177\u6709\u5927\u89c4\u6a21\u7a00\u758f\u7684\u7279\u70b9\u3002\u4e0d\u540c\u4e8eCV\u548cNLP\u573a\u666f\uff0c\u6570\u636e\u662f\u7a20\u5bc6\u7684\u56fe\u50cf\u548c\u6587\u672c\uff0c\u641c\u5e7f\u63a8\u7684\u6570\u636e\u975e\u5e38\u7a00\u758f\u7684\uff0c\u7b2c\u4e00\u8fd9\u6765\u6e90\u4e8e\u5f88\u591a\u6570\u636e\u65e0\u6cd5\u5bf9\u6240\u6709\u7528\u6237\u548c\u573a\u666f\u6709\u6548\u91c7\u96c6\u5230\uff0c\u7b2c\u4e8c\u662f\u56e0\u4e3a\u5efa\u6a21\u4f7f\u7528\u7684\u7279\u5f81\u91cf\u7ea7\u5927\u9020\u6210\u7684\u9ad8\u7ef4\u7a00\u758f\u6027\uff0c\u8fd9\u4f1a\u5f71\u54cd\u4e86\u6570\u636e\u7684\u5b58\u50a8\u683c\u5f0f\u548c\u8ba1\u7b97\u6548\u7387\u3002<\/li>\n<\/ol>\n<p>\u56e0\u6b64\uff0c\u89e3\u51b3CTR\u5927\u6a21\u578b\u7684\u8fd9\u79cd\u7a20\u5bc6\u53c2\u6570\u8f83\u5927\u7684\u6a21\u578b\uff0c\u5173\u952e\u662f\u5c06Sparse\u53c2\u6570\u7531\u5355\u673a\u5b58\u50a8\u6539\u9020\u4e3a\u5206\u5e03\u5f0f\u5b58\u50a8\uff0c\u5e76\u4e3b\u8981\u901a\u8fc7\u6570\u636e\u5e76\u884c\u63d0\u9ad8\u541e\u5410\u3002<\/p>\n<p>\u4e0b\u9762\u6211\u4eec\u8bf4\u660e\u4e0b\u5bf9\u8bad\u7ec3\u6846\u67b6\u4f18\u5316\u70b9\u3002\u6838\u5fc3\u7684\u4e24\u70b9\uff0c\u4e00\u4e2a\u5728\u4e8e\u5206\u5e03\u5f0f\u901a\u4fe1\u62d3\u6251\u7684\u8bbe\u8ba1\uff0c\u8fd8\u6709\u4e00\u4e2a\u5728\u4e8eEmbedding Lookup\u7684\u6027\u80fd\u4f18\u5316\u3002<\/p>\n<ol class=\"list-paddingleft-1\" data-id=\"odd3d8fc-oLGJWehT\">\n<li data-id=\"ld70c578-bdvAWq8v\">\u7a00\u758f\u53c2\u6570\uff0c\u501f\u52a9\u53c2\u6570\u670d\u52a1\u5668\uff08Param Server\uff09\uff0c\u5c06 embedding \u5b58\u50a8\u548c\u66f4\u65b0\u8d1f\u62c5\u8f6c\u5ac1\u5230PS\u3002\u7a00\u758f\u53c2\u6570 Partition \u5b58\u653e\uff0c\u6bcf\u4e2a Worker \u53ea\u6709\u90e8\u5206\u5206\u7247\uff0c\u68af\u5ea6\u66f4\u65b0\u65f6\u8fdb\u884c AlltoAll\uff0c\u60f3\u529e\u6cd5\u89e3\u51b3 \u7a00\u758ftensor \u7684\u5b58\u50a8\u3001\u901a\u4fe1\u6210\u672c\u3002<\/li>\n<li data-id=\"ld70c578-KrnnVq93\">\u7a20\u5bc6\u53c2\u6570\uff0c\u501f\u52a9\u4e8eallreduce\uff0c\u5c06\u7a20\u5bc6\u53c2\u6570 Replication \u5b58\u653e\uff0c\u6bcf\u4e2a Worker \u90fd\u6709\u526f\u672c\uff0c\u68af\u5ea6\u66f4\u65b0\u65f6\u8fdb\u884c allreduce\u3002allreduce \u548c alltoall \u90fd\u4f1a\u4f7f\u7528 nccl \u8fdb\u884c\u540c\u6b65\u901a\u4fe1\uff0c\u6548\u7387\u8f83\u9ad8\u3002hb \u8fdb\u884c alltoall \u65f6\uff0c\u901a\u4fe1\u7684\u662f\u7a00\u758f\u68af\u5ea6\uff0c\u800c\u4e0d\u662f\u5b8c\u6574\u7684\u53c2\u6570\uff0c\u901a\u4fe1\u91cf\u4e0a\u548c ps \u662f\u4e00\u81f4\u7684\uff0c\u4f46\u662f\u901a\u4fe1\u6548\u7387\u4f1a\u66f4\u9ad8\u3002<\/li>\n<\/ol>\n<p>\u5728\u5b9e\u73b0\u4e0a\uff0c\u53ef\u4ee5\u901a\u8fc7\u66ff\u6362TF\u539f\u751f\u7b97\u5b50\u8fdb\u884cSparse\u53c2\u6570\u7684\u8bfb\u53d6\u8fc7\u7a0b\uff08\u6838\u5fc3\u7b97\u5b50\u662fGatherV2\u7b97\u5b50\uff09\u3002<\/p>\n<p>\u8be5\u7b97\u5b50\u7684\u4f5c\u7528\u662f\u4eceEmbedding\u8868\u4e2d\u8bfb\u53d6ID\u5217\u8868\u7d22\u5f15\u5bf9\u5e94\u7684Embedding\u6570\u636e\u5e76\u8fd4\u56de\uff0c\u672c\u8d28\u4e0a\u662f\u4e00\u4e2aHash\u67e5\u8be2\u7684\u8fc7\u7a0b\uff1b<\/p>\n<p>\u901a\u8fc7\u66ff\u6362\u8be5\u7b97\u5b50\uff0c\u5e76\u540c\u65f6\u4fee\u6539\u4e0a\u4e0b\u6e38\u8282\u70b9\u7684Input\/Output\uff0c\u5c06\u4ece\u672c\u5730Embedding\u8868\u4e2d\u67e5\u8be2\uff0c\u6539\u9020\u4e3a\u4ece\u5206\u5e03\u5f0fKV\u4e2d\u67e5\u8be2\u3002<\/p>\n<h2>\u603b\u7ed3<\/h2>\n<p>\u53ef\u4ee5\u8bf4\uff0c\u8bad\u7ec3\u5927\u6a21\u578b\uff0c\u4e0d\u4ec5\u9700\u8981\u5145\u8db3\u7684\u8ba1\u7b97\u8d44\u6e90\u548c\u6570\u636e\uff0c\u8fd8\u9700\u8981\u6df1\u539a\u7ecf\u9a8c\u548c\u6280\u80fd\uff0c\u8fd8\u9700\u8981\u4e00\u5b9a\u7684\u8010\u5fc3\u548c\u5b9a\u529b\uff0c\u5c31\u50cf\u201c\u70bc\u4e39\u201d\u4e00\u6837\u3002<\/p>\n<p>\u6bcf\u6b21\u70bc\u4e39\u5e08\u7684\u51fa\u624b\u90fd\u6709\u7740\u5de8\u5927\u65f6\u95f4\u548c\u7ecf\u6d4e\u6210\u672c\uff0c\u5982\u4f55\u5728\u6700\u5c0f\u6210\u672c\u4e0b\u627e\u5230\u6700\u4f18\u89e3\uff0c\u5c31\u662f\u6211\u4eec\u4e00\u76f4\u5728\u63a2\u7d22\u7684\u3002<\/p>\n<p>\u6587\u7ae0\u6765\u81ea\uff1a51CTO<\/p>\n<\/div>\n<\/div>\n<div class=\"pvc_clear\"><\/div>\n<p id=\"pvc_stats_16223\" class=\"pvc_stats total_only  \" data-element-id=\"16223\" style=\"\"><i class=\"pvc-stats-icon medium\" aria-hidden=\"true\"><svg xmlns=\"http:\/\/www.w3.org\/2000\/svg\" version=\"1.0\" viewBox=\"0 0 502 315\" preserveAspectRatio=\"xMidYMid meet\"><g transform=\"translate(0,332) scale(0.1,-0.1)\" fill=\"\" stroke=\"none\"><path d=\"M2394 3279 l-29 -30 -3 -207 c-2 -182 0 -211 15 -242 39 -76 157 -76 196 0 15 31 17 60 15 243 l-3 209 -33 29 c-26 23 -41 29 -80 29 -41 0 -53 -5 -78 -31z\"\/><path d=\"M3085 3251 c-45 -19 -58 -50 -96 -229 -47 -217 -49 -260 -13 -295 52 -53 146 -42 177 20 16 31 87 366 87 410 0 70 -86 122 -155 94z\"\/><path d=\"M1751 3234 c-13 -9 -29 -31 -37 -50 -12 -29 -10 -49 21 -204 19 -94 39 -189 45 -210 14 -50 54 -80 110 -80 34 0 48 6 76 34 21 21 34 44 34 59 0 14 -18 113 -40 219 -37 178 -43 195 -70 221 -36 32 -101 37 -139 11z\"\/><path d=\"M1163 3073 c-36 -7 -73 -59 -73 -102 0 -56 133 -378 171 -413 34 -32 83 -37 129 -13 70 36 67 87 -16 290 -86 209 -89 214 -129 231 -35 14 -42 15 -82 7z\"\/><path d=\"M3689 3066 c-15 -9 -33 -30 -42 -48 -48 -103 -147 -355 -147 -375 0 -98 131 -148 192 -74 13 15 57 108 97 206 80 196 84 226 37 273 -30 30 -99 39 -137 18z\"\/><path d=\"M583 2784 c-38 -19 -67 -74 -58 -113 9 -42 211 -354 242 -373 16 -10 45 -18 66 -18 51 0 107 52 107 100 0 39 -1 41 -124 234 -80 126 -108 162 -133 173 -41 17 -61 16 -100 -3z\"\/><path d=\"M4250 2784 c-14 -9 -74 -91 -133 -183 -95 -150 -107 -173 -107 -213 0 -55 33 -94 87 -104 67 -13 90 8 211 198 130 202 137 225 78 284 -27 27 -42 34 -72 34 -22 0 -50 -8 -64 -16z\"\/><path d=\"M2275 2693 c-553 -48 -1095 -270 -1585 -649 -135 -104 -459 -423 -483 -476 -23 -49 -22 -139 2 -186 73 -142 361 -457 571 -626 285 -228 642 -407 990 -497 242 -63 336 -73 660 -74 310 0 370 5 595 52 535 111 1045 392 1455 803 122 121 250 273 275 326 19 41 19 137 0 174 -41 79 -309 363 -465 492 -447 370 -946 591 -1479 653 -113 14 -422 18 -536 8z m395 -428 c171 -34 330 -124 456 -258 112 -119 167 -219 211 -378 27 -96 24 -300 -5 -401 -72 -255 -236 -447 -474 -557 -132 -62 -201 -76 -368 -76 -167 0 -236 14 -368 76 -213 98 -373 271 -451 485 -162 444 86 934 547 1084 153 49 292 57 452 25z m909 -232 c222 -123 408 -262 593 -441 76 -74 138 -139 138 -144 0 -16 -233 -242 -330 -319 -155 -123 -309 -223 -461 -299 l-81 -41 32 46 c18 26 49 83 70 128 143 306 141 649 -6 957 -25 52 -61 116 -79 142 l-34 47 45 -20 c26 -10 76 -36 113 -56z m-2057 25 c-40 -58 -105 -190 -130 -263 -110 -324 -59 -707 132 -981 25 -35 42 -64 37 -64 -19 0 -241 119 -326 174 -188 122 -406 314 -532 468 l-58 71 108 103 c185 178 428 349 672 473 66 33 121 60 123 61 2 0 -10 -19 -26 -42z\"\/><path d=\"M2375 1950 c-198 -44 -350 -190 -395 -379 -18 -76 -8 -221 19 -290 114 -284 457 -406 731 -260 98 52 188 154 231 260 27 69 37 214 19 290 -38 163 -166 304 -326 360 -67 23 -215 33 -279 19z\"\/><\/g><\/svg><\/i> <img loading=\"lazy\" decoding=\"async\" width=\"16\" height=\"16\" alt=\"Loading\" src=\"https:\/\/aif.amtbbs.org\/wp-content\/plugins\/page-views-count\/ajax-loader-2x.gif\" border=0 \/><\/p>\n<div class=\"pvc_clear\"><\/div>\n","protected":false},"excerpt":{"rendered":"<p>\u53ef\u4ee5\u8bf4\uff0c\u8bad\u7ec3\u5927\u6a21\u578b\uff0c\u4e0d\u4ec5\u9700\u8981\u5145\u8db3\u7684\u8ba1\u7b97\u8d44\u6e90\u548c\u6570\u636e\uff0c\u8fd8\u9700\u8981\u6df1\u539a\u7ecf\u9a8c\u548c\u6280\u80fd\uff0c\u8fd8\u9700\u8981\u4e00\u5b9a\u7684\u8010\u5fc3\u548c\u5b9a\u529b\uff0c\u5c31\u50cf\u201c\u70bc\u4e39\u201d\u4e00 [&hellip;]<\/p>\n<div class=\"pvc_clear\"><\/div>\n<p id=\"pvc_stats_16223\" class=\"pvc_stats total_only  \" data-element-id=\"16223\" style=\"\"><i class=\"pvc-stats-icon medium\" aria-hidden=\"true\"><svg xmlns=\"http:\/\/www.w3.org\/2000\/svg\" version=\"1.0\" viewBox=\"0 0 502 315\" preserveAspectRatio=\"xMidYMid meet\"><g transform=\"translate(0,332) scale(0.1,-0.1)\" fill=\"\" stroke=\"none\"><path d=\"M2394 3279 l-29 -30 -3 -207 c-2 -182 0 -211 15 -242 39 -76 157 -76 196 0 15 31 17 60 15 243 l-3 209 -33 29 c-26 23 -41 29 -80 29 -41 0 -53 -5 -78 -31z\"\/><path d=\"M3085 3251 c-45 -19 -58 -50 -96 -229 -47 -217 -49 -260 -13 -295 52 -53 146 -42 177 20 16 31 87 366 87 410 0 70 -86 122 -155 94z\"\/><path d=\"M1751 3234 c-13 -9 -29 -31 -37 -50 -12 -29 -10 -49 21 -204 19 -94 39 -189 45 -210 14 -50 54 -80 110 -80 34 0 48 6 76 34 21 21 34 44 34 59 0 14 -18 113 -40 219 -37 178 -43 195 -70 221 -36 32 -101 37 -139 11z\"\/><path d=\"M1163 3073 c-36 -7 -73 -59 -73 -102 0 -56 133 -378 171 -413 34 -32 83 -37 129 -13 70 36 67 87 -16 290 -86 209 -89 214 -129 231 -35 14 -42 15 -82 7z\"\/><path d=\"M3689 3066 c-15 -9 -33 -30 -42 -48 -48 -103 -147 -355 -147 -375 0 -98 131 -148 192 -74 13 15 57 108 97 206 80 196 84 226 37 273 -30 30 -99 39 -137 18z\"\/><path d=\"M583 2784 c-38 -19 -67 -74 -58 -113 9 -42 211 -354 242 -373 16 -10 45 -18 66 -18 51 0 107 52 107 100 0 39 -1 41 -124 234 -80 126 -108 162 -133 173 -41 17 -61 16 -100 -3z\"\/><path d=\"M4250 2784 c-14 -9 -74 -91 -133 -183 -95 -150 -107 -173 -107 -213 0 -55 33 -94 87 -104 67 -13 90 8 211 198 130 202 137 225 78 284 -27 27 -42 34 -72 34 -22 0 -50 -8 -64 -16z\"\/><path d=\"M2275 2693 c-553 -48 -1095 -270 -1585 -649 -135 -104 -459 -423 -483 -476 -23 -49 -22 -139 2 -186 73 -142 361 -457 571 -626 285 -228 642 -407 990 -497 242 -63 336 -73 660 -74 310 0 370 5 595 52 535 111 1045 392 1455 803 122 121 250 273 275 326 19 41 19 137 0 174 -41 79 -309 363 -465 492 -447 370 -946 591 -1479 653 -113 14 -422 18 -536 8z m395 -428 c171 -34 330 -124 456 -258 112 -119 167 -219 211 -378 27 -96 24 -300 -5 -401 -72 -255 -236 -447 -474 -557 -132 -62 -201 -76 -368 -76 -167 0 -236 14 -368 76 -213 98 -373 271 -451 485 -162 444 86 934 547 1084 153 49 292 57 452 25z m909 -232 c222 -123 408 -262 593 -441 76 -74 138 -139 138 -144 0 -16 -233 -242 -330 -319 -155 -123 -309 -223 -461 -299 l-81 -41 32 46 c18 26 49 83 70 128 143 306 141 649 -6 957 -25 52 -61 116 -79 142 l-34 47 45 -20 c26 -10 76 -36 113 -56z m-2057 25 c-40 -58 -105 -190 -130 -263 -110 -324 -59 -707 132 -981 25 -35 42 -64 37 -64 -19 0 -241 119 -326 174 -188 122 -406 314 -532 468 l-58 71 108 103 c185 178 428 349 672 473 66 33 121 60 123 61 2 0 -10 -19 -26 -42z\"\/><path d=\"M2375 1950 c-198 -44 -350 -190 -395 -379 -18 -76 -8 -221 19 -290 114 -284 457 -406 731 -260 98 52 188 154 231 260 27 69 37 214 19 290 -38 163 -166 304 -326 360 -67 23 -215 33 -279 19z\"\/><\/g><\/svg><\/i> <img loading=\"lazy\" decoding=\"async\" width=\"16\" height=\"16\" alt=\"Loading\" src=\"https:\/\/aif.amtbbs.org\/wp-content\/plugins\/page-views-count\/ajax-loader-2x.gif\" border=0 \/><\/p>\n<div class=\"pvc_clear\"><\/div>\n","protected":false},"author":56,"featured_media":16224,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[23,80],"tags":[898],"class_list":["post-16223","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-23","category-80","tag-898"],"_links":{"self":[{"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/posts\/16223","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/users\/56"}],"replies":[{"embeddable":true,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/comments?post=16223"}],"version-history":[{"count":1,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/posts\/16223\/revisions"}],"predecessor-version":[{"id":16225,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/posts\/16223\/revisions\/16225"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/media\/16224"}],"wp:attachment":[{"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/media?parent=16223"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/categories?post=16223"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/aif.amtbbs.org\/index.php\/wp-json\/wp\/v2\/tags?post=16223"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}