kckrish21 commited on
Commit
87b1da9
·
verified ·
1 Parent(s): 221ad04

Add model card

Browse files
Files changed (1) hide show
  1. README.md +170 -0
README.md ADDED
@@ -0,0 +1,170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+
3
+ library_name: transformers
4
+ base_model: facebook/bart-large
5
+ pipeline_tag: text2text-generation
6
+ tags:
7
+ - bart
8
+ - title-generation
9
+ - scientific-text
10
+ - summarization
11
+ - scihigh-2026
12
+ - fire-2026
13
+
14
+
15
+ ---
16
+
17
+ # SciHigh 2026 Task 2 — BART-large Title Generation Model
18
+
19
+ This repository contains a fine-tuned **BART-large** model for **Task 2: Title Generation from Abstracts** of the SciHigh 2026 shared task at FIRE 2026.
20
+
21
+ The model takes a scientific paper abstract as input and generates a concise research-paper title.
22
+
23
+ ## Base Model
24
+
25
+ * Model: `facebook/bart-large`
26
+ * Architecture: BART encoder-decoder Transformer
27
+ * Parameters: approximately 406 million
28
+ * Framework: Hugging Face Transformers
29
+
30
+ ## Task
31
+
32
+ **Input:** Scientific paper abstract
33
+ **Output:** Generated scientific paper title
34
+
35
+ The model was fine-tuned for highly compressed abstractive generation, where the goal is to capture the central topic and contribution of an abstract in title form.
36
+
37
+ ## Dataset
38
+
39
+ The model was fine-tuned on the **SpringerSSAT** dataset released for SciHigh 2026 Task 2.
40
+
41
+ Dataset split used:
42
+
43
+ * Training: 2,778 abstract-title pairs
44
+ * Validation: 347 abstract-title pairs
45
+ * Test: 348 abstracts with masked reference titles
46
+
47
+ Only the training split was used for gradient updates. The validation set was used for evaluation and best-checkpoint selection.
48
+
49
+ ## Preprocessing
50
+
51
+ * Maximum abstract length: 512 tokens
52
+ * Maximum target-title length: 64 tokens
53
+ * Input truncation: enabled
54
+ * Target truncation: enabled
55
+ * Dynamic batch padding using `DataCollatorForSeq2Seq`
56
+ * No task-specific instruction prefix was added
57
+
58
+ ## Fine-Tuning Configuration
59
+
60
+ * Epochs: 3
61
+ * Learning rate: `3e-5`
62
+ * Weight decay: `0.01`
63
+ * Per-device training batch size: 1
64
+ * Per-device evaluation batch size: 1
65
+ * Gradient accumulation steps: 8
66
+ * Effective training batch size: 8
67
+ * Mixed-precision training: FP16 AMP
68
+ * Gradient checkpointing: enabled
69
+ * Evaluation strategy: once per epoch
70
+ * Checkpoint saving strategy: once per epoch
71
+ * Generation beam size during validation: 4
72
+ * Maximum generation length: 64
73
+ * Best-model selection metric: ROUGE-L
74
+ * Best model automatically restored after training
75
+
76
+ ## Validation Results
77
+
78
+ | Epoch | Validation Loss | ROUGE-1 | ROUGE-2 | ROUGE-L | ROUGE-Lsum |
79
+ | ----- | --------------: | ----------: | ----------: | ----------: | ----------: |
80
+ | 1 | 2.282688 | 44.9926 | 21.3081 | 36.6377 | 36.6539 |
81
+ | 2 | **2.217401** | **45.7446** | **22.2135** | **37.6433** | **37.6641** |
82
+ | 3 | 2.242662 | 45.2856 | 21.8391 | 37.2244 | 37.2773 |
83
+
84
+ ### Best Checkpoint
85
+
86
+ The best model was obtained at **Epoch 2**.
87
+
88
+ * ROUGE-1: **45.7446**
89
+ * ROUGE-2: **22.2135**
90
+ * ROUGE-L: **37.6433**
91
+ * ROUGE-Lsum: **37.6641**
92
+ * Validation loss: **2.217401**
93
+
94
+ A separate evaluation over all 347 validation examples reproduced these results.
95
+
96
+ These are validation-set results and are not official hidden-test scores.
97
+
98
+ ## Example Usage
99
+
100
+ ```python
101
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
102
+
103
+ model_name = "kckrish21/SciHigh2026-Task2-BART-large"
104
+
105
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
106
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
107
+
108
+ abstract = """
109
+ Insert a scientific paper abstract here.
110
+ """
111
+
112
+ inputs = tokenizer(
113
+ abstract,
114
+ return_tensors="pt",
115
+ max_length=512,
116
+ truncation=True
117
+ )
118
+
119
+ generated_ids = model.generate(
120
+ **inputs,
121
+ max_length=64,
122
+ num_beams=4,
123
+ early_stopping=True
124
+ )
125
+
126
+ title = tokenizer.decode(
127
+ generated_ids[0],
128
+ skip_special_tokens=True
129
+ )
130
+
131
+ print(title)
132
+ ```
133
+
134
+ ## Test-Set Generation
135
+
136
+ For SciHigh 2026 test inference:
137
+
138
+ * Test examples: 348
139
+ * Generation batch size: 4
140
+ * Beam size: 4
141
+ * Maximum generation length: 64
142
+ * Early stopping: enabled
143
+ * Missing predictions: 0
144
+ * Empty predictions: 0
145
+
146
+ The reference test titles are masked by the task organizers, so no test-set metric is reported here.
147
+
148
+ ## Intended Use
149
+
150
+ This model is intended for:
151
+
152
+ * scientific paper title generation from abstracts
153
+ * research on scientific text generation
154
+ * participation and reproducibility for SciHigh 2026 Task 2
155
+
156
+ Generated titles should be treated as model suggestions rather than authoritative paper titles.
157
+
158
+ ## Limitations
159
+
160
+ The model was fine-tuned on a relatively small dataset of social-science research abstracts. Performance may differ for scientific domains or writing styles that are substantially different from the SpringerSSAT training distribution.
161
+
162
+ Like other neural text-generation systems, the model may generate wording that is incomplete, overly generic, or not fully supported by the abstract.
163
+
164
+ ## Shared Task
165
+
166
+ **SciHigh 2026 — Research Highlight Generation from Scientific Papers**
167
+ **FIRE 2026**
168
+ Subtask 2: **Title Generation from Abstracts**
169
+
170
+ Ranking for the shared task is primarily based on ROUGE-L F1.