humanlong commited on
Commit
10aac26
Β·
verified Β·
1 Parent(s): 3d6c7b9

Initial model card (checkpoint coming soon)

Browse files
Files changed (1) hide show
  1. README.md +110 -0
README.md ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - negotiation
7
+ - emotion
8
+ - llm-agent
9
+ - lora
10
+ - peft
11
+ - offline-rl
12
+ - iql
13
+ - small-language-model
14
+ - edge-deployable
15
+ library_name: peft
16
+ base_model: Qwen/Qwen2.5-7B-Instruct
17
+ datasets:
18
+ - humanlong/emotion-negotiation-benchmarks
19
+ pipeline_tag: text-generation
20
+ ---
21
+
22
+ # EmoDistill-creditor-7b
23
+
24
+ **Offline-distilled 7B emotion-aware credit-recovery negotiation agent.**
25
+
26
+ EmoDistill turns a 7B base LLM into a domain-adaptive negotiation agent by decoupling *what emotion to show* from *how to express it*. It learns both from a fixed offline corpus of LLM-vs-LLM negotiations β€” **no online rollouts, no human feedback** β€” and refines the expression policy with a per-turn LLM judge.
27
+
28
+ This repo will host the released checkpoint and adapter weights for the EmoDistill 7B creditor agent. See the [code repository](https://github.com/Yunbo-max/EmoDistill) for training and inference.
29
+
30
+ > 🚧 **Status:** Pretrained 7B fine-tuned creditor checkpoint coming soon. This repo currently hosts the model card and configuration; the LoRA adapter and IQL emotion-selector weights will be uploaded once training finalizes.
31
+
32
+ ---
33
+
34
+ ## πŸ“ Method
35
+
36
+ EmoDistill composes **three offline-trained components** at inference:
37
+
38
+ 1. **IQL emotion selector** β€” Implicit Q-Learning over a **28-emotion vocabulary**, trained on logged LLM-vs-LLM negotiation trajectories. Picks the emotion to express at each turn.
39
+ 2. **LoRA-SFT expression imitation** β€” LoRA adapter on top of the 7B base, trained by *imitation* on top-K advantage-filtered offline turns. Learns to verbalize emotion-conditioned creditor utterances.
40
+ 3. **JPO (Judge Policy Optimization)** β€” PPO-clipped surrogate against a per-turn LLM judge, anchored by KL to the SFT init. Refines the LoRA adapter for naturalness and strategic effectiveness without destabilizing the SFT skills.
41
+
42
+ The three components are designed to be **fully offline** β€” no live LLM API needed at training time after the negotiation log is collected β€” and **edge-deployable**: the runtime is a single 7B model with a LoRA adapter and a small Q-network for emotion selection.
43
+
44
+ ## πŸš€ Intended use
45
+
46
+ - **Primary task:** automated, emotion-aware credit-recovery negotiation in agent-to-agent settings.
47
+ - **Deployment:** on-device / edge, where data-privacy constraints make calling a frontier LLM infeasible.
48
+ - **Base model:** [`Qwen/Qwen2.5-7B-Instruct`](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct). Compatible with both OpenAI-API and DashScope-API serving via the `LLMClient` wrapper in the [code repo](https://github.com/Yunbo-max/EmoDistill).
49
+
50
+ ## πŸ“Š Evaluation
51
+
52
+ The model is evaluated on the **credit_recovery** subset of [`humanlong/emotion-negotiation-benchmarks`](https://huggingface.co/datasets/humanlong/emotion-negotiation-benchmarks) (100 scenarios). Companion baselines for comparison:
53
+
54
+ - **[EQ-Negotiator](https://github.com/Yunbo-max/EQ-Negotiator)** (NeurIPS 2025, [arXiv:2511.03370](https://arxiv.org/abs/2511.03370)) β€” persona + HMM + WSLS, no learning.
55
+ - **[EmoMAS](https://github.com/Yunbo-max/EmoMAS)** (ACL 2026 Main, top 9%, [arXiv:2604.07003](https://arxiv.org/abs/2604.07003)) β€” Bayesian multi-agent orchestration, no pre-training.
56
+ - **[EvoEmo](https://github.com/Yunbo-max/EvoEmo)** (AAMAS 2026, [arXiv:2509.04310](https://arxiv.org/abs/2509.04310)) β€” online evolutionary emotion policies.
57
+ - Vanilla and fixed-emotion 7B baselines.
58
+
59
+ Headline results will be filled in here after the checkpoint upload.
60
+
61
+ ## πŸ“¦ Quick start (after checkpoint release)
62
+
63
+ ```python
64
+ from peft import PeftModel
65
+ from transformers import AutoModelForCausalLM, AutoTokenizer
66
+
67
+ base = "Qwen/Qwen2.5-7B-Instruct"
68
+ adapter = "humanlong/EmoDistill-creditor-7b"
69
+
70
+ tok = AutoTokenizer.from_pretrained(base)
71
+ model = AutoModelForCausalLM.from_pretrained(base, device_map="auto", torch_dtype="auto")
72
+ model = PeftModel.from_pretrained(model, adapter)
73
+
74
+ prompt = "<creditor system prompt with debtor context, target emotion: empathy>"
75
+ inputs = tok(prompt, return_tensors="pt").to(model.device)
76
+ out = model.generate(**inputs, max_new_tokens=200)
77
+ print(tok.decode(out[0], skip_special_tokens=True))
78
+ ```
79
+
80
+ For the full pipeline (IQL emotion selection β†’ LoRA generation β†’ JPO-refined responses), use the code in the [EmoDistill GitHub repo](https://github.com/Yunbo-max/EmoDistill).
81
+
82
+ ## ⚠️ Limitations
83
+
84
+ - Trained for **credit recovery** in English. Generalization to the other three domains (disaster, education, hospital) in the benchmark suite is not yet evaluated.
85
+ - The IQL emotion selector uses a fixed 28-emotion vocabulary; unseen emotions are not supported.
86
+ - The model is designed to be persuasive but ethical β€” adversarial use to manipulate vulnerable debtors is **out of scope** and explicitly discouraged.
87
+
88
+ ## πŸ“ License
89
+
90
+ Apache 2.0 β€” matches the base model.
91
+
92
+ ## πŸ“š Citation
93
+
94
+ ```bibtex
95
+ @inproceedings{emodistill2026,
96
+ title = {EmoDistill: Offline Emotion Skill Distillation for LM Negotiation Agents},
97
+ author = {Long, Yunbo and others},
98
+ year = {2026},
99
+ note = {Submitted to EMNLP}
100
+ }
101
+ ```
102
+
103
+ ## πŸ”— Related work β€” the full thread
104
+
105
+ | Work | Venue | Role |
106
+ |---|---|---|
107
+ | [EQ-Negotiator](https://github.com/Yunbo-max/EQ-Negotiator) | NeurIPS 2025 | Personas + HMM + WSLS for SLMs |
108
+ | [EvoEmo](https://github.com/Yunbo-max/EvoEmo) | AAMAS 2026 | Online evolutionary emotion policies |
109
+ | [EmoMAS](https://github.com/Yunbo-max/EmoMAS) | ACL 2026 (top 9%) | Bayesian multi-agent orchestration + 4 benchmarks |
110
+ | **EmoDistill** *(this repo)* | EMNLP submission | Offline distillation into 7B SLM |