summaryrefslogtreecommitdiff
path: root/configs
diff options
context:
space:
mode:
authorYurenHao0426 <Blackhao0426@gmail.com>2026-04-03 15:12:34 -0500
committerYurenHao0426 <Blackhao0426@gmail.com>2026-04-03 15:12:34 -0500
commit8fe28101366dd32562b8c5534d7fe359b252bdf3 (patch)
treec92a92184fb2f46f265ab84c1f754c3d5d6597bc /configs
Initial commit: UPH project codebase and experiment results
Includes model code, evaluation scripts, configs, analysis outputs, and experiment results for the User Prior Head personalization method. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Diffstat (limited to 'configs')
-rw-r--r--configs/__init__.py0
-rw-r--r--configs/review_temporal.yaml27
-rw-r--r--configs/review_user.yaml31
-rw-r--r--configs/topic_temporal.yaml27
-rw-r--r--configs/topic_user.yaml27
5 files changed, 112 insertions, 0 deletions
diff --git a/configs/__init__.py b/configs/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/configs/__init__.py
diff --git a/configs/review_temporal.yaml b/configs/review_temporal.yaml
new file mode 100644
index 0000000..8c7f56d
--- /dev/null
+++ b/configs/review_temporal.yaml
@@ -0,0 +1,27 @@
+task: review
+setting: temporal
+dataset_config: product_review_temporal
+model_name: Qwen/Qwen2.5-1.5B-Instruct
+device: cuda:1
+
+d: 64
+alpha: 0.1
+beta: 0.05
+lam: 0.0001
+lr: 0.05
+adapt_steps: 30
+max_grad_norm: 5.0
+basis_seed: 42
+
+K: 4
+
+max_new_tokens: 512
+temperature: 0.0
+top_p: 0.9
+
+num_eval: -1
+batch_size: 1
+
+# Blended generation
+blend_gamma: 0.5
+min_new_tokens: 128
diff --git a/configs/review_user.yaml b/configs/review_user.yaml
new file mode 100644
index 0000000..d2423ee
--- /dev/null
+++ b/configs/review_user.yaml
@@ -0,0 +1,31 @@
+task: review
+setting: user
+dataset_config: product_review_user
+model_name: Qwen/Qwen2.5-1.5B-Instruct
+device: cuda:1
+
+# CVH hyperparams
+d: 64
+alpha: 0.1
+beta: 0.05
+lam: 0.0001
+lr: 0.05
+adapt_steps: 30
+max_grad_norm: 5.0
+basis_seed: 42
+
+# Support set
+K: 4
+
+# Generation
+max_new_tokens: 512
+temperature: 0.0
+top_p: 0.9
+
+# Evaluation
+num_eval: -1 # -1 means all
+batch_size: 1
+
+# Blended generation
+blend_gamma: 0.5
+min_new_tokens: 128
diff --git a/configs/topic_temporal.yaml b/configs/topic_temporal.yaml
new file mode 100644
index 0000000..0af41d4
--- /dev/null
+++ b/configs/topic_temporal.yaml
@@ -0,0 +1,27 @@
+task: topic
+setting: temporal
+dataset_config: topic_writing_temporal
+model_name: Qwen/Qwen2.5-1.5B-Instruct
+device: cuda:1
+
+d: 64
+alpha: 0.1
+beta: 0.05
+lam: 0.0001
+lr: 0.05
+adapt_steps: 30
+max_grad_norm: 5.0
+basis_seed: 42
+
+K: 4
+
+max_new_tokens: 512
+temperature: 0.0
+top_p: 0.9
+
+num_eval: -1
+batch_size: 1
+
+# Blended generation
+blend_gamma: 0.5
+min_new_tokens: 128
diff --git a/configs/topic_user.yaml b/configs/topic_user.yaml
new file mode 100644
index 0000000..5dc7ac0
--- /dev/null
+++ b/configs/topic_user.yaml
@@ -0,0 +1,27 @@
+task: topic
+setting: user
+dataset_config: topic_writing_user
+model_name: Qwen/Qwen2.5-1.5B-Instruct
+device: cuda:1
+
+d: 64
+alpha: 0.1
+beta: 0.05
+lam: 0.0001
+lr: 0.05
+adapt_steps: 30
+max_grad_norm: 5.0
+basis_seed: 42
+
+K: 4
+
+max_new_tokens: 512
+temperature: 0.0
+top_p: 0.9
+
+num_eval: -1
+batch_size: 1
+
+# Blended generation
+blend_gamma: 0.5
+min_new_tokens: 128