shaswat123/NIAH
Retrieval Head This is the open-source code for paper: Retrieval Head Mechanistically Explains Long-Context Factuality. This code is implemented based on Needle In a HayStack. 【Update】 Support Phi3 now, thanks to the contribution made by @Wangmerlyn. Retrieval Head Detection An algorithm that statistically calculate the retrieval score of attention heads in a transformer model. Because FlashAttention can not return attention matrix, this algorithm is implemented… See the full description on the dataset page: https://huggingface.co/datasets/shaswat123/NIAH.
| Author / organization | shaswat123 |
|---|---|
| Downloads | 255 (as reported by Hugging Face) |
| Likes | 0 |
| Created | 30 June 2025 |
| Last updated at source | 30 June 2025 |
Popularity, last 3 days
255 now (+80 since 2026-10-16). Counts as reported by the source, snapshotted daily by Silytics.
Recent changes
- 17 Oct 2026 Description: Retrieval Head This is the open-source code for paper: Retrieval Head Mechanistically Explains Long-Context Factuality. This code is implemented based on Needle In a HayStack. 【Update】 Support Phi3 now, thanks to the contribution made by @W → Retrieval Head This is the open-source code for paper: Retrieval Head Mechanistically Explains Long-Context Factuality. This code is implemented based on Needle In a HayStack. 【Update】 Support Phi3 now, thanks to the contribution made by @W
Changes detected by Silytics between refreshes of the source.
More from shaswat123
Silytics shows public metadata from Hugging Face and links to the original. Information may be incomplete or out of date. Last refreshed 18 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.