GQA:GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints
arXiv · Google Research
提出分组查询注意力与通过均值池化和5%算力Uptrain的方案,以接近MQA的推理速度取得接近MHA的生成质量。
arXiv:2305.13245