Westlake-AI / MogaNet

[ICLR 2024] MogaNet: Efficient Multi-order Gated Aggregation Network
https://arxiv.org/abs/2211.03295
Apache License 2.0
162 stars 13 forks source link

Could you share the code for drawing multi-order interactions that shown in Fig. 3 of your paper? #4

Closed VoXuanThuy closed 1 year ago

VoXuanThuy commented 1 year ago

Thank you for the great work. I have a concern about how to draw the figure of the multi-order interactions. Could you share the code? Thank you in advance.

Jacky1128 commented 1 year ago

Hi, @VoXuanThuy, thanks for your attention. You can refer to Section B.1 of MogaNet for more implementation details of multi-order interaction. As for the source code, please refer to Representation Bottleneck and draw_figures.py for the interaction figure drawing. I hope this will be helpful to you and feel free to ask me if you have more problems with MogaNet.

Lupin1998 commented 1 year ago

Hi, @VoXuanThuy. We hope the official implementation of Representation Bottleneck could help you. Here, we also upload a modified implementation version (RepBottleneck-main.zip) for our experiments (please do not distribute). You can ask me or contact the author of Representation Bottleneck if some questions occur. I will close this issue if there is no more question.