SearcharxivSearch

arXiv subjects

Kaishun Zhang

Publications and source records attributed to Kaishun Zhang.

2 recordsLinked to original sources

Fixed-Attention Mechanism for Deep-Learning-Assisted Design of High-Degree-of-Freedom 3D Metamaterials

The traditional design approaches for high-degree-of-freedom metamaterials have been computationally intensive and, in many cases, even intractable due to the vast design space. In this work, we introduce a novel fixed-attention mechanism into a deep learning framework to address the computational challenges of metamaterial design. We consider a 3D plasmonic structure composed of gold nanorods characterized by geometric parameters and demonstrate that a Long Short-Term Memory network with a fixed-attention mechanism can improve the prediction accuracy by 48.09% compared to networks without attention. Additionally, we successfully apply this framework for the inverse design of plasmonic metamaterials. Our approach significantly reduces computational costs, opening the door for efficient real-time optimization of complex nanostructures.

physics.optics

MiniMax-01: Scaling Foundation Models with Lightning Attention

We introduce MiniMax-01 series, including MiniMax-Text-01 and MiniMax-VL-01, which are comparable to top-tier models while offering superior capabilities in processing longer contexts. The core lies in lightning attention and its efficient scaling. To maximize computational capacity, we integrate it with Mixture of Experts (MoE), creating a model with 32 experts and 456 billion total parameters, of which 45.9 billion are activated for each token. We develop an optimized parallel strategy and highly efficient computation-communication overlap techniques for MoE and lightning attention. This approach enables us to conduct efficient training and inference on models with hundreds of billions of parameters across contexts spanning millions of tokens. The context window of MiniMax-Text-01 can reach up to 1 million tokens during training and extrapolate to 4 million tokens during inference at an affordable cost. Our vision-language model, MiniMax-VL-01 is built through continued training with 512 billion vision-language tokens. Experiments on both standard and in-house benchmarks show that our models match the performance of state-of-the-art models like GPT-4o and Claude-3.5-Sonnet while offering 20-32 times longer context window. We publicly release MiniMax-01 at https://github.com/MiniMax-AI.

cs.CL