[GH-ISSUE #29] About expert capacities: Is there token-dropping during training? #26

Closed
opened 2026-10-26 21:01:49 +00:00 by bsakaguchi · 3 comments
Owner

Originally created by @Spico197 on GitHub (Mar 4, 2024).
Original GitHub issue: https://github.com/deepseek-ai/DeepSeek-MoE/issues/29

Hi there, thanks for open-sourcing such an amazing LM~

I'm wondering if you limit the maximum expert capacity by dropping some tokens during training like Switch Transformer.
Since training large MoE models is pretty costly, maybe it's necessary to add a hard constraint to ensure faster expert parallelism?

Thanks again and looking forward to hearing from you.

image

Originally created by @Spico197 on GitHub (Mar 4, 2024). Original GitHub issue: https://github.com/deepseek-ai/DeepSeek-MoE/issues/29 Hi there, thanks for open-sourcing such an amazing LM~ I'm wondering if you limit the maximum expert capacity by dropping some tokens during training like Switch Transformer. Since training large MoE models is pretty costly, maybe it's necessary to add a hard constraint to ensure faster expert parallelism? Thanks again and looking forward to hearing from you. ![image](https://github.com/deepseek-ai/DeepSeek-MoE/assets/22840952/6a2c7216-1dfb-476a-8cdf-65a14dfb3b8f)
Author
Owner

@zwd003 commented on GitHub (Apr 3, 2024):

@DeepSeekDDM

<!-- gh-comment-id:2034703272 --> @zwd003 commented on GitHub (Apr 3, 2024): @DeepSeekDDM
Author
Owner

@DeepSeekDDM commented on GitHub (Apr 3, 2024):

DeepSeekMoE 16B does not have a dropping strategy since it does not use expert parallelism. If using expert parallelism, token dropping will be applied to ensure training efficiency.

<!-- gh-comment-id:2035053064 --> @DeepSeekDDM commented on GitHub (Apr 3, 2024): DeepSeekMoE 16B does not have a dropping strategy since it does not use expert parallelism. If using expert parallelism, token dropping will be applied to ensure training efficiency.
Author
Owner

@Spico197 commented on GitHub (Apr 7, 2024):

Thanks for your valuable comments~ I found the relevant contents from the technical report. Thanks again for your response~

image

<!-- gh-comment-id:2041311811 --> @Spico197 commented on GitHub (Apr 7, 2024): Thanks for your valuable comments~ I found the relevant contents from the technical report. Thanks again for your response~ ![image](https://github.com/deepseek-ai/DeepSeek-MoE/assets/22840952/cdfa8af4-b08a-4386-80e9-04e57ca95966)
Sign in to join this conversation.
No labels
pull-request
WIP
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepSeek-MoE#26
No description provided.