allenai/soda

Dataset

157

stars

4

commits

2

linked in READMEs

Jan 4, 2023

updated

commonsense
dialogue
narrative

README

Dataset Card for 🥤SODA

Dataset Description

Dataset Summary

🥤SODA is the first publicly available, million-scale, high-quality dialogue dataset covering a wide range of social interactions. Dialogues are distilled from a PLM (InstructGPT; Ouyang et al., 2022) by contextualizing social commonsense knowledge from a knowledge graph (Atomic10x; West et al., 2022). Human evaluation shows that dialogues in SODA are more consistent, specific, and (surprisingly) natural than prior human-authored datasets – e.g., DailyDialog (Li et al., 2017), BlendedSkillTalk (Smith et al., 2020). Also, since social commonsense knowledge encompasses emotional reactions (i.e., the xReact relation), SODA includes 385K conversations labeled with 1.7K unique emotions along with information about the experiencer and the cause – i.e., PersonX and the head event in the symbolic commonsense knowledge triple.

Languages

English

Dataset Structure

fieldtypedescription
headstrthe head event in the symbolic commonsense knowledge triple
relationstrthe relationship between head and tail events
tailstrthe tail event in the symbolic commonsense knowledge triple
literalstrthe symbolic commonsense knowledge in sentence-form
narrativestrnarrative based on the literal
dialoguelist of strdialogue grounded in the narrative
speakerslist of strthe speakers for each turn in the dialogue
PersonXstrthe assigned name for PersonX in the commonsense knowledge triple
PersonYstr|nullthe assigned name for PersonY in the commonsense knowledge triple
PersonZstr|nullthe assigned name for PersonZ in the commonsense knowledge triple
original_indexintthe original index from Atomic10x
splitstrthe split information: {train, valid, test}
head_answerstrthe answer for whether the head is included in the narrative: {Yes, Unknown}
pmi_head_answerstrthe answer for whether the head is included in the narrative with point-wise mutual information applied: {Yes, No, Unknown}
relation_tail_answerstrthe answer for whether the relation-tail is included in the dialogue: {Yes, No, Unknown}
pmi_relation_tail_answerstrthe answer for whether the relation-tail is included in the dialogue with point-wise mutual information applied: {Yes, No, Unknown}

Dataset Creation

To create 🥤SODA, we distill dialogues from InstructGPT by contextualizing social commonsense knowledge – i.e., adding context information in multiple steps: (1) Retrieve social commonsense from the symbolic commonsense knowledge graph, (2) convert it into sentence form, (3) generate a narrative from the sentence, (4) infer the speakers from the narrative, and finally (5) derive contentful conversation grounded in the narrative and speakers. Anchoring the PLM in commonsense knowledge for deriving conversations offers two key advantages: (1) minimizing nonsensical conversations and (2) maximizing diversity. For more details, please refer to our paper.

Further Details, Social Impacts, and Limitations

Please refer to our paper.

Trained Model

Using 🥤SODA, we train 🧑🏻‍🚀COSMO: a generalizable conversation agent outperforming previous best-performing agents on both in- and out-of-domain datasets. COSMO-3B is available here!

Additional Information

For a brief summary of our paper, please see this tweet.

Citation

Please cite our work if you find the resources in this repository useful:

@article{kim2022soda,
    title={SODA: Million-scale Dialogue Distillation with Social Commonsense Contextualization},
    author={Hyunwoo Kim and Jack Hessel and Liwei Jiang and Peter West and Ximing Lu and Youngjae Yu and Pei Zhou and Ronan Le Bras and Malihe Alikhani and Gunhee Kim and Maarten Sap and Yejin Choi},
    journal={ArXiv},
    year={2022},
    volume={abs/2212.10465}
}

Contributors

heanu

3 commits

HK
Hyunwoo Kim

1 commits

allenai/soda

Dataset

157

stars

4

commits

2

linked in READMEs

Jan 4, 2023

updated

commonsense
dialogue
narrative

README

Dataset Card for 🥤SODA

Dataset Description

Dataset Summary

🥤SODA is the first publicly available, million-scale, high-quality dialogue dataset covering a wide range of social interactions. Dialogues are distilled from a PLM (InstructGPT; Ouyang et al., 2022) by contextualizing social commonsense knowledge from a knowledge graph (Atomic10x; West et al., 2022). Human evaluation shows that dialogues in SODA are more consistent, specific, and (surprisingly) natural than prior human-authored datasets – e.g., DailyDialog (Li et al., 2017), BlendedSkillTalk (Smith et al., 2020). Also, since social commonsense knowledge encompasses emotional reactions (i.e., the xReact relation), SODA includes 385K conversations labeled with 1.7K unique emotions along with information about the experiencer and the cause – i.e., PersonX and the head event in the symbolic commonsense knowledge triple.

Languages

English

Dataset Structure

fieldtypedescription
headstrthe head event in the symbolic commonsense knowledge triple
relationstrthe relationship between head and tail events
tailstrthe tail event in the symbolic commonsense knowledge triple
literalstrthe symbolic commonsense knowledge in sentence-form
narrativestrnarrative based on the literal
dialoguelist of strdialogue grounded in the narrative
speakerslist of strthe speakers for each turn in the dialogue
PersonXstrthe assigned name for PersonX in the commonsense knowledge triple
PersonYstr|nullthe assigned name for PersonY in the commonsense knowledge triple
PersonZstr|nullthe assigned name for PersonZ in the commonsense knowledge triple
original_indexintthe original index from Atomic10x
splitstrthe split information: {train, valid, test}
head_answerstrthe answer for whether the head is included in the narrative: {Yes, Unknown}
pmi_head_answerstrthe answer for whether the head is included in the narrative with point-wise mutual information applied: {Yes, No, Unknown}
relation_tail_answerstrthe answer for whether the relation-tail is included in the dialogue: {Yes, No, Unknown}
pmi_relation_tail_answerstrthe answer for whether the relation-tail is included in the dialogue with point-wise mutual information applied: {Yes, No, Unknown}

Dataset Creation

To create 🥤SODA, we distill dialogues from InstructGPT by contextualizing social commonsense knowledge – i.e., adding context information in multiple steps: (1) Retrieve social commonsense from the symbolic commonsense knowledge graph, (2) convert it into sentence form, (3) generate a narrative from the sentence, (4) infer the speakers from the narrative, and finally (5) derive contentful conversation grounded in the narrative and speakers. Anchoring the PLM in commonsense knowledge for deriving conversations offers two key advantages: (1) minimizing nonsensical conversations and (2) maximizing diversity. For more details, please refer to our paper.

Further Details, Social Impacts, and Limitations

Please refer to our paper.

Trained Model

Using 🥤SODA, we train 🧑🏻‍🚀COSMO: a generalizable conversation agent outperforming previous best-performing agents on both in- and out-of-domain datasets. COSMO-3B is available here!

Additional Information

For a brief summary of our paper, please see this tweet.

Citation

Please cite our work if you find the resources in this repository useful:

@article{kim2022soda,
    title={SODA: Million-scale Dialogue Distillation with Social Commonsense Contextualization},
    author={Hyunwoo Kim and Jack Hessel and Liwei Jiang and Peter West and Ximing Lu and Youngjae Yu and Pei Zhou and Ronan Le Bras and Malihe Alikhani and Gunhee Kim and Maarten Sap and Yejin Choi},
    journal={ArXiv},
    year={2022},
    volume={abs/2212.10465}
}

Contributors

heanu

3 commits

HK
Hyunwoo Kim

1 commits