Constrained Multi-Layer Contrastive Learning for Implicit Discourse Relationship Recognition

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Yiheng, Li, Junhui, Zhu, Muhua
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917781352480768
author Wu, Yiheng
Li, Junhui
Zhu, Muhua
author_facet Wu, Yiheng
Li, Junhui
Zhu, Muhua
contents Previous approaches to the task of implicit discourse relation recognition (IDRR) generally view it as a classification task. Even with pre-trained language models, like BERT and RoBERTa, IDRR still relies on complicated neural networks with multiple intermediate layers to proper capture the interaction between two discourse units. As a result, the outputs of these intermediate layers may have different capability in discriminating instances of different classes. To this end, we propose to adapt a supervised contrastive learning (CL) method, label- and instance-centered CL, to enhance representation learning. Moreover, we propose a novel constrained multi-layer CL approach to properly impose a constraint that the contrastive loss of higher layers should be smaller than that of lower layers. Experimental results on PDTB 2.0 and PDTB 3.0 show that our approach can significantly improve the performance on both multi-class classification and binary classification.
format Preprint
id arxiv_https___arxiv_org_abs_2409_13716
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Constrained Multi-Layer Contrastive Learning for Implicit Discourse Relationship Recognition
Wu, Yiheng
Li, Junhui
Zhu, Muhua
Computation and Language
Previous approaches to the task of implicit discourse relation recognition (IDRR) generally view it as a classification task. Even with pre-trained language models, like BERT and RoBERTa, IDRR still relies on complicated neural networks with multiple intermediate layers to proper capture the interaction between two discourse units. As a result, the outputs of these intermediate layers may have different capability in discriminating instances of different classes. To this end, we propose to adapt a supervised contrastive learning (CL) method, label- and instance-centered CL, to enhance representation learning. Moreover, we propose a novel constrained multi-layer CL approach to properly impose a constraint that the contrastive loss of higher layers should be smaller than that of lower layers. Experimental results on PDTB 2.0 and PDTB 3.0 show that our approach can significantly improve the performance on both multi-class classification and binary classification.
title Constrained Multi-Layer Contrastive Learning for Implicit Discourse Relationship Recognition
topic Computation and Language
url https://arxiv.org/abs/2409.13716