Lighthouse Attention is a pretty clever idea, first coarse filtering then fine calculation, splitting long texts into short segments fed into FlashAttention, achieving 17 times faster on a single B200 card with 512K. During the final training stages, switch back to full attention to prevent skipping reads, and the 5.3B model still maintains performance. Long context optimization finally no longer requires stubbornly optimizing underlying kernels.

View Original
Original content no longer visible
This page may contain third-party content, which is provided for information purposes only (not representations/warranties) and should not be considered as an endorsement of its views by Gate, nor as financial or professional advice. See Disclaimer for details.
  • Reward
  • Comment
  • Repost
  • Share
Comment
Add a comment
Add a comment
No comments
  • Pinned