From 08a08befe8ac0202ce952b88aa144cbffe2a73f9 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 11 Feb 2026 00:21:34 +0000 Subject: Add dual-vector ablation experiment (z_short/z_long necessity) - Add rag_vector_no_short (eta_short=0) and rag_vector_no_long (eta_long=0) configs - Add ablation run script for 60p60s experiment - Document ablation design and expected results in notes Co-Authored-By: Claude Opus 4.6 --- .../adapters/personalized_llm_adapter.py | 34 ++++++++++++++++++++++ 1 file changed, 34 insertions(+) (limited to 'collaborativeagents/adapters/personalized_llm_adapter.py') diff --git a/collaborativeagents/adapters/personalized_llm_adapter.py b/collaborativeagents/adapters/personalized_llm_adapter.py index 488b241..4189f31 100644 --- a/collaborativeagents/adapters/personalized_llm_adapter.py +++ b/collaborativeagents/adapters/personalized_llm_adapter.py @@ -672,6 +672,40 @@ def create_baseline_adapter( "extractor": "cuda:2", }, ), + # Ablation: RAG + Vector without z_short (only z_long, no within-session adaptation) + "rag_vector_no_short": AdapterConfig( + mode="full", + enable_preference_extraction=True, + enable_rl_updates=True, + use_user_vector=True, + llm_name=llm_name, + use_shared_models=use_shared_models, + enable_query_transform=True, + enable_global_preferences=True, + eta_short=0.0, # Disable z_short learning + device_assignment={ + "embed": "cuda:2", + "reranker": "cuda:3", + "extractor": "cuda:2", + }, + ), + # Ablation: RAG + Vector without z_long (only z_short, no cross-session learning) + "rag_vector_no_long": AdapterConfig( + mode="full", + enable_preference_extraction=True, + enable_rl_updates=True, + use_user_vector=True, + llm_name=llm_name, + use_shared_models=use_shared_models, + enable_query_transform=True, + enable_global_preferences=True, + eta_long=0.0, # Disable z_long learning + device_assignment={ + "embed": "cuda:2", + "reranker": "cuda:3", + "extractor": "cuda:2", + }, + ), # Baseline 7a: RAG + Vector + Preference Rewrite (combines best of both) "rag_rewrite_vector": AdapterConfig( mode="full", -- cgit v1.2.3