Skip to content

Citation, and bidirectional attention results on a smaller decoder - #5

Merged
akanyaani merged 1 commit into
mainfrom
readme-citation
Oct 23, 2026
Merged

akanyaani merged 1 commit into
mainfrom
readme-citation

Conversation

@akanyaani

Copy link
Copy Markdown
Contributor
  • Citation: the README now ends with a BibTeX entry. A new CITATION.cff (validated against CFF 1.2.0) makes GitHub show a "Cite this repository" button. The author is listed as FrontiersMind AI, matching the package metadata; individual names can be added.
  • Bidirectional attention on a smaller decoder: three 50-step LoRA runs of the deciformer15k_qwen3 recipe (Qwen3-0.6B, deciformer-15k, 256 dev records), identical except for attention and pooling:
attention pooler train loss, step 10 → 50 dev accuracy
causal last token 1.55 → 0.73 0.672
bidirectional last token (Decision 3.0 layout) 2.89 → 1.27 0.449
bidirectional mean 2.63 → 1.25 0.441

The weak bidirectional result is not specific to Qwen3.5: a causally pretrained decoder switched to bidirectional attention starts from a much worse point and learns far more slowly. The results are added to docs/benchmarks.md, and the README's attention section quotes them.

…Qwen3-0.6B

- README ends with a BibTeX entry; CITATION.cff (valid CFF 1.2.0) gives
  GitHub's "Cite this repository".
- docs/benchmarks.md: 50-step comparison on deciformer-15k with Qwen3-0.6B:
  causal 0.672, bidirectional with the last-token readout 0.449, with a mean
  pooler 0.441. The README's attention section quotes it.
@akanyaani
akanyaani merged commit 9dccd66 into main Oct 23, 2026
3 checks passed
akanyaani added a commit that referenced this pull request Oct 23, 2026
Pre-release fixes from the developer review (includes #5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant