Alexandria-Gemma-4-E4B-it-Qwen27-Knowledge-Units-LoRA-r128

Project Alexandria — Gemma 4 E4B IT: Knowledge Units, LoRA rank 128#

This is a LoRA adapter, released on 10 October 2026. Load it on the pinned Gemma 4 E4B IT base model; it contains learned adapter matrices rather than a standalone copy of Gemma.

Why extract scientific knowledge?#

Research papers contain findings, experimental conditions, quantities and relationships that agents can use for retrieval, question answering and synthesis. Project Alexandria aims to make that factual knowledge reusable through representations that reduce dependence on the original prose. A Knowledge Unit describes entities, attributes and relationships; a scientific summary offers an organized account of a paper.

The motivation is introduced in Project Alexandria: Towards Freeing Scientific Knowledge from Copyright Burdens via LLMs. The paper proposes knowledge-preserving, style-agnostic representations and examines their reuse. An agent can retrieve relevant KUs, answer a question from their facts and follow a source identifier for verification. Distillation trains a smaller extractor to imitate a larger teacher, reducing the cost of creating these representations.

For example, an illustrative KU can describe an intervention, its dose, the measured effect, the population and the uncertainty as separate fields. This helps an agent distinguish a measured result from an interpretation. These adapters learn the extraction task; they do not provide a database of all scientific knowledge.

The technical goal is to preserve scientific information while reducing reproduction of source expression. This release does not certify that its outputs are copyright-free. Our literal-copy audits still find substantial overlap. Source access, source terms, attribution and output review remain relevant. QA accuracy measures usefulness for answering questions; it does not establish legal clearance or complete factual correctness.

Model and training provenance#

• Base: google/gemma-4-E4B-it, revision ee0ef6023621cff504d758262d4e04895a5af4a2.
• Teacher: Qwen/Qwen3.8-27B-FP8, revision 017b9c7af6b5689d5dd426a76e0bc077eb5ca20a.
• Task: generator-only knowledge units; no quality-review or correction adapter was trained.
• Training: 391 examples from 97 papers, one epoch, 49 optimizer updates.
• LoRA: rank 128, alpha 256, dropout 0.05; language-layer q/k/v/o attention and gate/up/down MLP projections.
• Optimizer: AdamW, learning rate 2e-5, weight decay 0.01, 5% warmup then cosine decay; BF16 base, gradient checkpointing, flex attention and assistant-token-weighted cut cross entropy.
• Effective batch: eight; source/prompt tokens are masked from the loss; no source or target truncation.
• See training configuration, native data manifest and weight/code provenance.

The teacher processed the 97-paper cohort sequentially with at most 1,000 source words per call, conditioning naming on the previous ten KUs. The teacher thinking mode was disabled, so these targets contain no reasoning supervision. The adapter was then evaluated on 20 different papers, with 200 questions frozen before context generation. It was applied to both 1,000-word and 500-word extraction without retraining. The 97 training papers are not an adapter test set.

The completed E4B training worker took 487.41 seconds on one GH200; optimizer computation accounted for 460.20 seconds. The adapter contains 279,052,288 trainable parameters. These worker timings exclude allocation startup and unrelated tasks.

Evaluation: retained knowledge#

A separate, fixed Qwen2.5-7B-Instruct answerer receives the generated representation and answers four-choice questions. The generator never receives evaluation questions or gold labels. Each paper contributes ten questions. Failed paper generations count as ten wrong answers; invalid responses remain in the denominator. Confidence intervals use 10,000 paired paper-bootstrap draws.

Matched 20-paper / 200-question follow-up#

This adapter reaches 82.50% with 500-word chunks and 80.00% with 1,000-word chunks. One failed 500-word paper generation remains in the denominator as ten wrong answers. The +2.5-point chunk difference has a paired 95% interval of −10.0 to +12.5 points. The small cohort does not establish a reliable advantage of 500-word extraction. All 18 context conditions were freshly judged in the follow-up, with unchanged cached contexts preserved. Questions were teacher-authored and source-validated automatically, rather than independently curated by humans.

The complete follow-up used four GH200 GPUs for 1,078 allocation seconds: 1.1978 reserved GPU-hours, including server startup, four Gemma extraction variants, all 3,600 QA slots and copying audits. This is an experimental allocation cost, not a production rate.

Copy-overlap audit#

Every evaluated output is audited against its complete, untruncated source. The primary rule allows at most five normalized consecutive whitespace words; six is borderline and seven or more is flagged. Narrative fields, evidence quotes and bibliographic metadata are tracked separately; a punctuation-split diagnostic is also retained. This numerical heuristic is a technical diagnostic, not a legal test.

No evaluated output from this adapter passes the strict five-word narrative rule under these recipes. No back-translation or semantic correction is applied in the primary scores. Raw QA performance and copying compliance are reported independently. The inference script performs the same audit automatically and retains failures and partial outputs.

Full English evaluation report and local machine-readable metrics, copy audit.

Two chunk sizes, one trained adapter#

There is one set of rank-128 adapter weights, trained with 1,000-word teacher chunks. The 500-word profile and 1,000-word profile apply that same adapter at inference. Use --chunk-words 500 or --chunk-words 1000 in the CLI below. No separate 500-word training run is implied.

The corresponding Gemma 4 12B KU adapter scores 88.50% at 500 words and 85.00% at 1,000 words on these same frozen 20 papers.

Quick start: inference#

Download this repository and install the inference environment separately from the training environment. All commands below run from the downloaded repository directory. A CUDA installation compatible with your vLLM build is required.

hf download laion/Alexandria-Gemma-4-E4B-it-Qwen27-Knowledge-Units-LoRA-r128 --local-dir adapter-release
cd adapter-release
python -m venv .venv-infer
. .venv-infer/bin/activate
python -m pip install -r requirements-inference.txt
vllm serve google/gemma-4-E4B-it --revision ee0ef6023621cff504d758262d4e04895a5af4a2 \
  --served-model-name base --host 127.0.0.1 --port 8000 \
  --language-model-only --reasoning-parser gemma4 \
  --enable-lora --max-lora-rank 128 --max-loras 2 \
  --lora-modules alexandria=. --generation-config vllm \
  --gpu-memory-utilization 0.90 --max-num-seqs 16 \
  --max-model-len 32768 --max-num-batched-tokens 8192 \
  --enable-prefix-caching --enable-chunked-prefill

In another terminal, use the same environment:

python code/infer.py --text-file paper.txt --task ku \
  --model alexandria --output outputs --chunk-words 500
# Batch input: one JSON object per line with document_id, text, optional title.
python code/infer.py --input papers.jsonl --task ku \
  --model alexandria --concurrency 16 --output outputs-batch --chunk-words 1000

Each document receives an output directory with complete call traces, the result, failure flags and the full-source copying audit. Chunks within a paper remain sequential; concurrent papers are batched by vLLM. KU generation disables thinking and allows one 16,384-token format retry after an 8,192-token attempt. Summary generation allows three bounded 24,576-token format attempts. Removing --thinking enables direct summary generation. Portable examples retain the benchmark prompts and budgets; hardware, scheduler ordering and library differences can change sampled outputs. They do not claim byte-identical regeneration of the archived benchmark.

Reproduce training#

The included compressed JSONL preserves the native tokenizer input IDs, assistant-only labels and original row order used for the completed run. Its uncompressed SHA256 is in the manifest. These token arrays encode source text and targets; they are reconstructable, so they are not privacy-preserving or irreversible fingerprints. Source-paper rights still apply.

python -m venv .venv-train
. .venv-train/bin/activate
python -m pip install -r requirements-training.txt
python code/train.py --output reproduced --dry-run
python code/train.py --output reproduced

The original KU run used one GH200 with microbatch one and gradient accumulation eight. Original summary runs used eight GH200 GPUs across two nodes, one example per rank. The portable trainer supports any GPU world size dividing eight. For two nodes with four GPUs each, launch torchrun --nnodes=2 --nproc_per_node=4 --node_rank=NODE_RANK --master_addr=MASTER_HOST --master_port=29500 code/train.py --output reproduced on each node. Adapter training needs substantial GPU memory for the 65,536-token summary limit; the original KU examples are much shorter.

code/prepare_training.py can render new OpenAI-style source-only conversations with actual reasoning_content through the pinned native tokenizer. It derives assistant boundaries from template offsets, preserves emitted reasoning and rejects over-length examples instead of truncating them. The released native arrays already provide the exact training targets. code/merge_adapter.py optionally merges these matrices into a downloaded BF16 base for deployment; quantization is a separate step.

License, attribution and citation#

• Release-owned adapter matrices, new scripts in code/, documentation and new metadata: CC BY 4.0. Attribute LAION / Project Alexandria contributors, link this model repository and identify your modifications.
• Bundled original Alexandria utilities and prompts in vendor/ and prompts/: retain their Apache 2.0 license and attribution; see third-party notices and Apache text. New release wrappers are CC BY 4.0.
• The Google base model and its tokenizer retain their own Apache 2.0 terms. The base model weights are obtained separately and are not redistributed here.
• Source documents and pre-existing datasets retain their own rights. This release licenses the contributions we can license; it does not relicense source-paper expression or guarantee downstream output clearance.

Please cite the Alexandria paper, base model and this adapter repository. Citation metadata and source-code provenance are included.

Release validation#

The packaged trainer validates all native examples and supervision masks. CPU checks compare one source conversation from each training-data family with its original input IDs and labels, inspect every adapter tensor shape, and verify the synthetic HTTP inference-client and copying-audit path. The completed GPU evaluations establish the performance of the original weights. No new full training run was launched to test packaging. Validation records.

Knowledge Units after guarded back-translation#

Read five complete papers and compare model/chunk views · Saved factual examples · PDF evaluation.

An additional matched study reused the same frozen 20 papers / 200 MCQs and freshly judged all 18 raw/reference contexts plus ten guarded KU contexts. The adapter weights and original extractions are unchanged. Copied context-summary/descriptive-attribute passages receive up to two TranslateGemma 4B English → German → English round-trips; graph names, IDs, targets and metadata are preserved. Repairs are accepted before QA using critical-value signatures, bidirectional NLI ≥0.9 and a five-word copy limit inside each candidate window.

These fresh stochastic QA scores are distinct from the earlier 20-paper results above. Failed generations retain their full wrong/invalid slots. No output is regenerated or selected using QA. Both comparisons have paired intervals including zero. Guarded document-level number/unit/formula checks record zero retained critical-value flags, but substantial literal overlap remains; the whole-output five-word rule is not completely met. The paper/graph/source references and missing author/style/lookahead annotations remain inspectable in the reader.

All new QA, fidelity and timing metrics, copy audit, separate bibliographic copy diagnostic and tested repair/evaluation code are included. The repair bundle needs its own runtime models/local source arrays as described by the linked experiment; training and ordinary adapter inference use the existing code above.

Full experiment documentation.

Wording repair: measured results and deployment advice#

Both KU chunk profiles are available#

Use 500 source words or 1000 source words with the same trained rank128 adapter. Training used 1000-word teacher chunks; these are two evaluated inference configurations, not two separately trained adapters.

The new extensive repair search below uses the 12B KU generator. Its scores are reference results, not new repair measurements for this E4B generator. The E4B model is also used without a KU adapter as the separate local editor.

The targeted editor is Gemma-4-E4B-it, BF16 with thinking disabled; “E4B” is the model identifier, not a measured 4-bit quantization setting. The tested repair chain contains TranslateGemma, Qwen local/full-field editing, and two Gemma local-edit stages. Direct Gemma-only repair is a proposed ablation whose complete throughput and QA have not been measured.

Twenty frozen papers, 200 questions per condition; 44 comparison conditions and 8,800 QA slots in the complete search. Every question/paper slot is retained. Bibliographic metadata is reported separately; the historical all-factual-field audit passes only 2/20 and 4/20. No tested pipeline achieves zero five-word-rule violations. The repeated holdout search still needs independent validation.

For the Qwen-rank128 merged-FP8 summary deployment, one guarded TranslateGemma round changes QA from 94.23% to 93.30% on 97 papers / 970 MCQs and lowers copied-word coverage from 15.23% to 9.69%, but all 97 narratives still violate the strict rule. Back-translation alone is a cheap option for partial overlap reduction; it is not validated as sufficient for complete compliance. Optimized generation plus that translation/NLI stage is estimated at 72,488 / 114,454 GH200 GPU-hours for 38M / 60M summaries, assuming 85% useful capacity.

The corresponding direct-Gemma cost scenarios are 108,683 / 171,605 GPU-hours for summary generation plus repair, and 232,315 / 366,813 for 500-word KU generation plus repair. These totals borrow 3.1296 GPU-seconds per paper from Gemma’s residual KU edits after earlier repair stages. Summary application and direct repair from raw KUs are unconfirmed optimistic proxies, not measured complete repair costs, guaranteed budgets or zero-overlap estimates. Generation-only estimates are 69,820 / 110,241 for summaries and 193,451 / 305,449 for 500-word KUs, with the same capacity factor.

Repair workflow and exact editor settings, full results, all scores, overlap evidence, 38M/60M compute assumptions, and frozen protocol are included. Scan the complete source, edit only flagged fields, preserve graph references and numbers/formulas, use bidirectional meaning filters against the immutable original, bound retries, and retain unresolved failures. QA is never an acceptance filter.

GitHub experiment and adapter inventory.

Related Alexandria adapters#

• Qwen27-Summaries-LoRA-r64.
• Qwen27-Summaries-LoRA-r128.
• Ornith9-Summaries-LoRA-r64.
• Ornith9-Summaries-LoRA-r128.
• Qwen27-Knowledge-Units-LoRA-r128.