Abstract
Text representation is a fundamental cornerstone that impacts the effectiveness of several text summarization methods. Transfer learning using pre-trained word embedding models has shown promising results. However, most of these representations do not consider the order and the semantic relationships between words in a sentence, and thus they do not carry the meaning of a full sentence. To overcome this issue, the current study proposes an unsupervised method for extractive multi-document summarization based on transfer learning from BERT sentence embedding model. Moreover, to improve sentence representation learning, we fine-tune BERT model on supervised intermediate tasks from GLUE benchmark datasets using single-task and multi-task fine-tuning methods. Experiments are performed on the standard DUC’2002–2004 datasets. The obtained results show that our method has significantly outperformed several baseline methods and achieves a comparable and sometimes better performance than the recent state-of-the-art deep learning–based methods. Furthermore, the results show that fine-tuning BERT using multi-task learning has considerably improved the performance.
Keywords
1. Introduction
In today’s society, we are facing an inevitable and challenging problem of information overload that highlighted the need of developing effective and specific tools to deal with this problem. Automatic text summarization (ATS), by condensing and creating shortened versions of texts, can efficiently help users save their time and effort while finding concise and suitable information for their needs. Thus, the main idea of ATS is to find a subset of data that capture the core information of the entire set. Different types of automatic text summarization have been proposed. For instance, regarding the number of input documents, single and multi-document summarization have been introduced. Similarly, based on the purpose of the summarization, it can be either generic or query-focused. Generic summaries represent all relevant facts of a source document without considering the users’ information needs, whereas in query-focused summaries, the content of the summary is driven by the user’s information need or simply the user’s query. In this article, we focus on generic multi-document summarization.
Since the pioneering work of Luhn [1], several methods have been proposed for automatic text summarization. They are mainly classified into extractive and abstractive approaches. The former generates summaries by extracting the most salient sentences as they exactly appear in the source documents, while abstractive summarization techniques produce summaries by concisely paraphrasing the document’s content. Abstractive methods are considered complex and sometimes they are not completely automatic; they require resources previously built that demand a high computational effort. Thus, as part of our research, we will use the extractive approach for generic multi-document summarization.
Document representation is considered as a fundamental process that affects the effectiveness of many extractive text summarization methods. For instance, Bag-of-Words (BOW) representation has been widely used in automatic text summarization [2]. However, the latter representation suffers from two major drawbacks: it does not take into consideration the ordering of words and ignores the semantic relationships between them. Although Bag-of-N-grams representation considers the order of the words in a short context, it suffers from data sparsity and curse of dimensionality problems [3]. Recently, word embeddings such as Word2vec [4] and GloVe [5] have been emerged as an effective representational basis for text summarization methods [6–10]. They aim to represent words as dense vectors in low-dimensional vector space using various pre-trained models, inspired from neural networks language modelling. Even though word embeddings have shown good performance in several natural language processing tasks [11,12], representing relationships among multiple words and phrases in a single dense vector is still an emerging problem. For example, considering these two sentences ‘you are going there to play not to study’ and ‘you are going there to study not to play’, these two sentences will have identical representation based on word embedding and BOW representations while their meanings are completely different.
In order to deal with the abovementioned problems, several sentence embedding models have been developed to represent variable-length sentences by dense vectors in a low dimensional vector space, which capture the semantic relationships among their constituents. We distinguish between two popular approaches for learning sentence embeddings, namely multi-task learning and language model pre-training. On the one hand, Multi-Task Learning (MTL) is an inductive mechanism that aims to improve generalisation performance by getting the benefit of the domain information contained in training signals of related tasks [13]. For two fundamental reasons, applying MTL to sentence representation learning using deep neural networks has shown promising results [14–18]. First, the limited amount of task-specific training data that is typically required to train any supervised deep neural network. Second, MTL benefits from regularisation effect that avoids over-fitting to a specific task, and hence, providing universal representations across tasks. On the other hand, the idea behind the language model pre-training approach is to first train deep neural network models on a large amount of unlabeled data using unsupervised objectives. Then, apply these pre-trained models to other supervised natural language understanding tasks using task-specific data. This allows building general-purpose models that have the ability to transfer the learned knowledge to other similar tasks as well as to help in saving time and computational power. For instance, pre-trained sentence embedding models such as the Bidirectional Encoder Representations from Transformers (BERT) [19], Skip-Thoughts [20] and ELMo [21] have shown to be effective for learning universal sentences representations, which are useful for many natural language processing tasks including automatic text summarization methods [22].
In this article, we propose an unsupervised extractive method for multi-document summarization based on transfer learning from the fine-tuned BERT models. We combine the strengths of both multi-task learning and the pre-trained language model BERT [19] to encode the input documents and obtain their sentences representation. We argue that these two technologies are complementary to each other and can be combined for improving sentence representation learning to boost the performance of text summarization task.
To summarise, the main contributions of this work are as follows:
We propose an unsupervised method for extractive multi-document summarization based on transfer learning from BERT fine-tuning on Natural Language Understanding tasks for sentence representation learning.
We fine-tune BERT model on supervised intermediate tasks from GLUE benchmark [23] using single-task and multi-task fine-tuning.
We investigate the impact of the two BERT fine-tuning methods on extractive multi-document summarization, and we showcase that BERT multi-task fine-tuning achieves substantial performance improvement.
To demonstrate the effectiveness of the proposed method, we empirically evaluate its performance on the standard DUC’2002–2004 multi-document summarization datasets, using the state-of-the-art ROUGE method [24]. Specifically, ROUGE-N (ROUGE-1, ROUGE-2, ROUGE-4) and ROUGE-L metrics aim to measure the content similarity between the generated summaries and their corresponding reference summaries (gold summaries). The obtained results show that the proposed method significantly outperforms several baseline methods and is on a par with recent state-of-the-art methods for extractive multi-document summarization. Moreover, the proposed method is unsupervised, fast and easy to implement.
The remainder of this article is organised as follows. We review the related work in section 2. We describe the proposed method in section 3. The experimental results on benchmark datasets are presented in section 4 and discussed in section 5. Finally, section 6 concludes the paper and draws lines for future works.
2. Related work
In this section, we first review some previous methods for extractive text summarization. Then, we provide a brief overview of the transfer learning methods applied in text representation learning. In addition finally, we cover BERT model architecture.
2.1. Extractive text summarization
Extractive text summarization methods identify the most salient sentences of a document and then subsequently concatenate them as they appear in the original document to create the final summary. Traditional text summarization methods are rule-based; they rely on hand-crafted features and expert knowledge [25–29]. With the recent advancements of neural network architectures, text summarization methods based on deep neural networks have received much attention and have achieved promising results. In the rest of this section, a brief description of extractive text summarization methods based on deep learning models is given. For readers who are interested in a detailed overview of automatic text summarization approaches and methods, they may refer to the recent surveys on the field [30,31].
In Zhong et al. [32], the authors have proposed an unsupervised method for multi-document summarization based on a deep learning model with three layers: concepts extraction, summary generation and reconstruction validation. Cao et al. [33] have used a recursive neural network to automatically rank sentences for multi-document summarization. While Yasunaga et al. [34] have introduced a neural multi-document summarization system (GBN-MDS) based on both graph convolutional networks and recurrent neural networks. Besides, Lebanoff et al. [35] have proposed to adapt encoder–decoder models trained on single-document datasets to the multi-document summarization task by introducing an external maximal marginal relevance module to select relevant sentences from multi-document input.
Moreover, deep neural networks have been also exploited to learn word/sentence representations for extractive text summarization. Word embedding representations have been proven to be effective for improving the performances of several text summarization methods [6–10]. The representation power of neural networks is related to their ability to learn high-level features across multiple layers and create accurate decision boundaries for the input instances. Following this success, other works have explored the potential of deep learning models for sentence representation learning. In particular, convolutional neural networks models have been widely used to learn sentence representations for extractive text summarization. For instance, Denil et al. [36] have developed a hierarchical convolutional model to introspect the structure of the document where a convolutional network is applied to learn sentences representations based on their words embeddings. In a similar work, Yin and Pei [37] have introduced a convolutional neural network–based method, where each sentence is projected to a continuous vector space, and then an optimization process is run to select relevant sentences taking into consideration their diversity and prestige cost. In the same context, Cao et al. [38] have developed a system based on enhanced convolutional neural networks that aims to automatically learn summary’s prior features for extractive summarization task.
In recent years, sentence embedding models have been widely used for extractive text summarization task [6,39,40]. Sentence embedding models aim to map sentences into dense vectors that encode their semantics. Zhang et al. [41] have proposed a sentence vector encoding framework based on a deep LSTM model; it embeds sentences into continuous vectors for single-line text summarization task. Joshi et al. [22] have proposed an unsupervised method for extractive text summarization based on the Skip-Thoughts pre-trained sentence embedding model [20]. In the same context, Bouscarrat et al. [42] have introduced STRASS, another extractive text summarization method based on sentence embedding representations. Where, the summary is created by selecting the sentences that are more similar to the entire embedding of the original document. Recently, Liu and Lapata [43] have developed a general framework for both extractive and abstractive text summarization based on the Bidirectional Encoder Representations from Transformers (BERT) [19], where authors have introduced a novel-based BERT document-level encoder able to capture the semantics of a document and thus generate representations of its sentences.
In contrast to the existing methods, we introduce a simple, effective and unsupervised method for extractive generic multi-document summarization. We explore transfer learning from BERT sentence embedding model [19] to improve sentence representation learning, which helps boost the performance of the proposed method. Inspired by human activities where people often apply the knowledge learned from previous tasks to help learn a new task, we fine-tune BERT on intermediate natural language understanding tasks from GLUE datasets before applying it to the text summarization task. Our idea is justified by the fact that transfer learning allows benefitting from knowledge learned from other natural language understanding tasks. Moreover, rather than fine-tuning BERT on a single task, we propose to fine-tune BERT on multiple related tasks simultaneously, which helps learn more universal sentence representation. Therefore, in this work, we showcase how the combination of transfer and multi-task learning approaches can be helpful in learning suitable sentence representation for extractive multi-document summarization. To the best of our knowledge, this is the first work that exploits the potential of BERT model fine-tuning on natural language understanding tasks (GLUE datasets) for unsupervised multi-document summarization task.
2.2. Transfer learning
Transfer learning is considered as a promising strategy that allows leveraging knowledge learned from one or more related tasks to boost the performance of a target task. It plays a key role in many natural language processing tasks, more specifically in text representations learning. Most of the transfer learning models are based on deep neural networks where there are two ways to apply neural network–based transfer learning methods for text representation learning: parameter initialization and multi-task learning. In some cases, parameter initialization and multi-task learning methods are used together to construct precise target models.
In the case of parameter initialization approach, a deep neural network is first trained on a source task, and then, the learned parameters are used to initialise the target task neural network model. In fact, two methods exist for applying the initialization parameter approach, including parameters freezing and fine-tuning methods. The freezing method uses the model trained on the source task data to extract features in order to use them as input to the target task without any modification; it is mainly helpful when there is a limited labelled data for the target task. While the fine-tuning method trains the neural network model in a source task data, the pre-trained model is applied to the target task where parameters of some layers are fixed and the others are learned on the target data. Therefore, with the success of distributed representations, pre-trained language models such as ELMo [21], BERT [19], Skip-Thoughts [20] and InferSent [44] have demonstrated promising benefits in learning universal sentence representations.
Nevertheless, most of the existing text representation models have been trained using single tasks such as predicting the next word or sentence [4,20] or text entailment [44]. Hence, the use of a small amount of task-specific labelled data can affect the performance of these models because supervised deep neural networks require a large amount of labelled data, which is not always available. To overcome this issue, multi-task learning approach proposes to learn text representations using supervised data from multiple related tasks. The use of this approach allows benefitting from a regularisation effect that helps in reducing the risk of over-fitting, as well as making the learned representations universal across tasks. Several works have been proposed for learning text representation using multi-task learning [14,15,16,18]. For instance, Liu et al. [14] have proposed a novel method for learning text representation across multiple tasks. The proposed method aims to leverage large amounts of task-specific data, as well as it benefits from a regularisation effect that leads to learn universal and useful representations for other new tasks. Luong et al. [15] have proposed to combine multi-task learning and sequence to sequence learning to learn suitable text representations for machine translation task. Moreover, Hashimoto et al. [45] have presented a joint many-task model with growing depth in a single end-to-end model, which makes use of linguistic hierarchies to solve increasingly complex natural language processing tasks. Jernite et al. [46] have developed a novel unsupervised method for learning sentence representation inspired by the notion of discourse coherence. The proposed method combines three auxiliary tasks to train the neural network model, namely the binary ordering of sentences task, the newt sentence prediction task and the conjunction prediction task. Furthermore, Guo et al. [16] have developed an abstractive method for document summarization based on a novel multi-task learning architecture that combines two auxiliary tasks including question generation and entailment generation tasks. Recently, Ruder et al. [18] have introduced a novel method for learning text representation based on latent multi-task learning.
In addition to the initialization parameter and the multi-task learning approaches, other works have been proposed to combine the strengths of both of them. For instance, the Multi-Task Deep Neural Networks (MT-DNN) model [47] has recently emerged as a new technology for learning contextual text representations among multiple natural language understanding related tasks. The MT-DNN differs from the traditional pre-trained language models in that, instead of using only pre-training to learn text representations, MT-DNN model adds multi-task objectives in order to learn more general and pertinent representations. Moreover, Liu et al. [48] have introduced the distilled Multi-Task Deep Neural Networks model (MT-DNNKD) using the same architecture as that of the MT-DNN model; however, the former is trained using the knowledge distillation method [49] in the multi-task learning settings.
2.3. BERT model architecture
The Bidirectional Encoder Representations from Transformers (BERT) model [19] is based on a multi-layer bidirectional transformer [50] with attention mechanisms. BERT is trained on large amount of source data (about 3300M words) from English Wikipedia and BookCorpus [51], using two unsupervised tasks including the masked language modelling and the next sentence prediction. Then, the pre-trained model can be applied to a new natural language processing task by adding a few layers to the source model, such as text classification [52], question/answering systems [19] and automatic text summarization [43]. The original architecture of BERT model is illustrated in Figure 1. Given the input

Architecture of the original BERT model [43]. The sequence on the top represents the input document, namely a sentence
Where
3. Proposed method
In this section, we present our method for generic extractive multi-document summarization. It consists of two main steps: BERT fine-tuning and multi-document summarization. In the first step, we fine-tune BERT model using both single-task and multi-task learning from the GLUE benchmark datasets [23]. In the second step, the fine-tuned BERT models are used to extract sentence representations for text summarization. Based on the latter sentences representation, a score is assigned to each sentence in the cluster by linearly combing three sentence scoring measures (Sentence Content Relevance, Sentence Novelty and Sentence Position Scores). The flowchart of the proposed method is shown in Figure 2. In what follows, we present the two steps of the proposed method (cf. sections 3.1 and 3.2).

Architecture of the proposed method for extractive multi-document summarization.
3.1. BERT fine-tuning
Since we introduce an unsupervised method for extractive multi-document summarization, we benefit from transfer learning abilities. In order to generate an embedding vector for each sentence of the input documents, we fine-tune the pre-trained BERT model on intermediate tasks, namely GLUE benchmark tasks, before applying it in text summarization task. We fine-tune BERT model using two methods, including BERT single-task fine-tuning and BERT multi-task fine-tuning. In the rest of this section, we present in detail the two BERT fine-tuning methods.
3.1.1. BERT Single-Task Fine-Tuning
In this step, the pre-trained model BERT is fine-tuned for each GLUE task using task-specific data. We use a model-based transfer learning approach, where the pre-trained model BERT is applied to a specific task by adding additional few layers to the source model with parameters of some layers stay the same while parameters of other layers are learned using the new task-specific data. In the following, we describe briefly the tasks used to fine-tune the pre-trained BERT model.
Single-Sentence Classification Task aims to predict a class label
where
Pairwise Text Classification Task: taking as an example the Natural Language Inference (NLI) task, given a pair of sentences
Text Similarity Task: given a pair of sentences
where
Relevance Ranking Task: We take as an example the Stanford Question Answering (QNLI) task [56]. Given a pair of a question and its candidate answer
where
3.1.2. BERT Multi-Task Fine-Tuning
In this step, we use the MT-DNN method [46] for BERT multi-task learning. We fine-tune the pre-trained BERT model simultaneously on four Natural Language Understanding tasks, including the single-sentence classification, pair-wise text classification, text similarity and relevance ranking, where each task has its own task-specific output layers (described in the previous section 3.1.1). The pre-trained BERT model is first used to initialise the parameters of the shared layers. Then, a mini-batch-based stochastic gradient descent is used to learn the model parameters, namely the parameters of shared layers and task-specific output layers. In detail, as shown in Algorithm 1 [48], the training samples from multiple tasks are packed into mini-batches denoted by
3.2. Multi-document summarization
Extractive multi-document summarization task is defined as the process of selecting the most relevant sentences that represent the content of the entire cluster of documents. It consists of three main steps: sentence embedding, sentence scoring and sentence selection.
3.2.1. Sentence Embedding
Let denote
3.2.2. Sentence Scoring
In this step, we assign a score for each sentence
Sentence Content Relevance Score: formally, given a cluster of documents
Second, we compute the cosine similarity between each sentence embedding vector
where
Sentence Novelty Score: since we address in this work the multi-document summarization task, redundancy represents a critical problem. The risk to select sentences that convey the same information is more prominent in contrast to single-document summarization. Thus, in order to deal with redundancy and produce summaries with good information diversity, we use the sentence novelty metric [22]. It assigns a low score to the sentence when it is redundant and a high score when it is novel. Hence, to get the novelty score of a sentence
Then, if the maximum of the obtained similarities
where
Sentence Position Score has been frequently applied in automatic text summarization [27,22], and it is considered as one of the effective methods for selecting relevant sentences, especially in newswire documents. The idea behind the sentence position hypothesis is that the first sentences of a document represent the most pertinent ones, and the importance decreases while we get further away from the beginning of the document. The sentence position in our case is used as a complementary handcrafted metric for sentence scoring. Formally, given
where
3.2.3. Sentence Selection
Finally, in order to get the final score of a sentence
where,
4. Experimental results
In this section, we present a comparative analysis of the results obtained using the proposed method. Several experiments have been conducted to address the following issues: (1) Investigating the use of BERT fine-tuning using single-task as well as multi-task learning for sentence representation learning. (2) Determining the natural language understanding tasks that allow transferring useful knowledge to extractive text summarization task. In addition, (3) assessing the performance of the proposed method in contrast to the baseline and state-of-the-art methods. However, before presenting the experimental results, we provide a brief description of the datasets used, the evaluation measures, the experimental setup and the approaches used for comparative analysis.
4.1. Datasets
We evaluate the proposed method on three standard datasets for generic multi-document summarization, namely DUC (Document Understanding Conference) (2002-2004) datasets, created by the National Institute of Standards and Technology (NIST). DUC’2002 and DUC’2003 consist, respectively, of 59 and 30 clusters where each cluster contains approximately 10 English newswire articles, distributed by TREC. Besides, Task 2 of DUC’2004 dataset comprises 50 clusters, where each cluster includes 10 documents, coming from the Associated Press and New York Times newswires. Table 1 summarises some basic statistics of these datasets.
A description of DUC’2002, DUC’2003 and DUC’2004 datasets [55].
DUC: document understanding conference.
For BERT fine-tuning, we have used the General Language Understanding Evaluation (GLUE) benchmark [23]; it is considered well-designed for evaluating the performance and generalisation of Natural Language Understanding (NLU) models that share general linguistic knowledge across a diverse set of related tasks. The GLUE benchmark represents a collection of multiple NLU tasks, including single-sentence classification, pairwise text classification, text similarity and relevance ranking, as summarised in Table 2. As well, it includes diverse datasets that are described in the following:
Corpus of Linguistic Acceptability (CoLA) [57] and Stanford Sentiment Treebank (SST-2) [53] are both single binary sentence classification tasks. The main aim of CoLA is to predict the linguistic plausibility of an English sentence. While SST-2 aims to classify the sentiment of a set of sentences extracted from movies reviews to positive and negative ones.
The Semantic Textual Similarity Benchmark (STS-B) [55] represents a regression task that predicts the semantic similarity score between a pair of sentences.
Recognising Textual Entailment (RTE) [58] and Multi-Genre Natural Language Inference (MNLI) [59] are both language inference tasks. Given a pair of sentences, the objective is to predict whether the second sentence is an entailment, contradiction or neutral according to the first one.
Quora Question Pairs 3 (QQP) dataset and Microsoft Research Paraphrase Corpus (MRPC) [60] are pairwise text classification tasks; the goal is to predict if two sentences are semantically equivalent.
Question Answering Language Inference (QNLI) is a version of the Stanford Question Answering Dataset (SQuAD) [56] that is defined as a binary classification task in GLUE. Given a query, the goal is to predict the relevant answer to this query.
Descriptions and statistics of GLUE tasks [23].
GLUE: general language understanding evaluation; CoLA: corpus of linguistic acceptability; SST: semantic textual similarity; RTE: recognising textual entailment; NLI: Natural Language Inference; MNLI: Multi-Genre Natural Language Inference; QQP: Quora Question Pairs; MRPC: Microsoft Research Paraphrase Corpus; STS-B: Semantic Textual Similarity Benchmark; QNLI: Question Answering Language Inference.
4.2. Evaluation measures
For evaluation, we have adopted the widely used method in evaluating automatic text summarization systems, namely ROUGE (Recall-Oriented Understudy for Gisting Evaluation) [24], particularly ROUGE-N (ROUGE-1 (R-1), ROUGE-2 (R-2) and ROUGE-4 (R-4)). ROUGE, as it is defined in equation (10), measures the similarity between system summaries and a collection of summaries models (human summaries) based on the n-gram comparison and overlap. We have calculated ROUGES scores with the ROUGE toolkit (version 1.5.5), adopting the same ROUGE settings 4 that are used on DUC datasets for generic extractive multi-document summarization
where
where
4.3. Experimental setup
The proposed method has been developed using Python and based on the PyTorch implementation of BERT model, 5 Multi-Task Deep Neural Networks (MT-DNN) model, 6 and Transformers. 7
For a complete comparison, we explore the potential of BERT model using three approaches: (1) feature-based approach, (2) BERT single-task fine-tuning and (3) BERT multi-task fine-tuning. In the feature-based approach, fixed features are extracted using the pre-trained encoder BERT and directly used as input features for the task at hand. In the second approach, a simple output layer is added to the pre-trained model, where all the parameters are jointly fine-tuned using task-specific data. In the third approach, BERT is fine-tuned simultaneously on four natural language processing tasks. All the experiments were performed based on tasks from GLUE benchmark datasets using BERT BASE and BERT LARGE . BERT BASE model is designed to embed a sentence into 768-dimensional vectors, while BERT LARGE provides sentence embeddings vectors of 1024 dimensions.
For BERT single-task fine-tuning, the only new parameters are those of the new additional output layer related to the downstream task. Most hyper-parameters are the same as in the pre-training step [19], except for the batch size, number of training epochs and the learning rate. Generally, the optimal values of these hyper-parameters depend on task-specific data; however, we found a possible range of values that work well across all tasks. For BERT BASE , we use a batch size of 32, 3 epochs and a learning rate of 5e-5 for all the GLUE tasks. For BERT LARGE , we use a batch size of 16, 5 epochs and a learning rate of 5e-5 for all the GLUE task. For BERT Multi-Task fine-tuning, we followed the same setup of the MT-DNN model [46]. We used Adamax optimizer [61] with a learning rate of 5e-5 and a batch size of 32 and 16 for BERT BASE and BERT LARGE , respectively. The number of epochs was set to 3 for BERT BASE and 5 for BERT LARGE .
All our experiments as well as BERT fine-tuning have been performed using an Intel(R) Xeon(TM) CPU 3.00 GHz server equipped with Nvidia Tesla K40c GPU having 12 Go of RAM. The checkpoints of the fine-tuned BERT models have been stored and used in an off-line mode to generate the embeddings vectors for all the input clusters sentences contained in DUC’2002–2004 datasets. It is worth mentioning that BERT models Fine-Tuning time depends on the task at hand and its size (BASE or LARGE). For instance, the time of
Execution time in minute for summarising each dataset DUC’2002–2004 using the fine-tuned BERT BASE and BERT LARGE models.
DUC: document understanding conference; BERT: bidirectional encoder representations from transformers.
As we have mentioned in section 3, the final score of a sentence
4.4. Results
First, the experiments are conducted to investigate the impact of using transfer learning from BERT model to improve the task of extractive multi-document summarization. The main goal of these experiments is to answer the following research question: Given the recent successes of BERT fine-tuning on intermediate-tasks, which tasks allow to transfer useful knowledge to extractive multi-document summarization?
To answer the latter question, we have applied BERT on the three summarization datasets DUC (2002–2004) by adopting two approaches, described as follows:
Approach 1: we have used the pre-trained BERT BASE and BERT LARGE models with no fine-tuning (feature-based approach);
Approach 2: we have fine-tuned BERT BASE and BERT LARGE models for each GLUE task on task-specific data (CoLA, SST-2, RTE, MNLI, QQP, MRPC, STS-B and QNLI);
Table 4 illustrates the overall obtained results on the three multi-document summarization datasets DUC’2002, DUC’2003 and DUC’2004 in terms of ROUGE-1, ROUGE-2 and ROUGE-4 recall scores. The results show clearly that applying BERT BASE and BERT LARGE models on extractive multi-document summarization without fine-tuning leads to poor performance on the three DUC (2002–2004) datasets. These results can be due to the fact that BERT model is trained on a masked language model where the output vectors are grounded to tokens instead of sentences, while most of extractive multi-document summarization methods manipulate sentence-level representations. Moreover, the overall obtained results demonstrate that fine-tuning BERT model yields significant improvements over its pre-trained model (without fine-tuning). For instance, the R-1 score on DUC’2004 has been improved from 28.02% to 39.04% after fine-tuning BERT BASE using the RTE dataset of the GLUE benchmark.
Comparison results of the different tasks used for BERT Fine-Tuning on the three extractive multi-document summarization datasets.
DUC: document understanding conference; BERT: bidirectional encoder representations from transformers; SST: semantic textual similarity; RTE: recognising textual entailment; MNLI: Multi-genre natural language inference; QQP: quora question pairs; MRPC: microsoft research paraphrase corpus; STS-B: semantic textual similarity benchmark; QNLI: question answering language inference.
For denoting statistical significance results, the superscript numbers indicate significant improvement (p value <0.05) over the task that has the same superscript number attached. The interval
For DUC’2002 dataset, and based on R-1 measure,
For DUC’2003 corpus, the results show that
For DUC’2004 dataset, and based on the three scores(R-1, R-2 and R-4), the obtained results show that fine-tuning BERT-BASE and
To summarise, the overall obtained results on the three DUC (2002–2004) datasets demonstrated that fine-tuning the pre-trained BERT model on coLA, SST-2, RTE, MNLI and QNLI datasets allows transferring useful and suitable knowledge for extractive multi-document summarization. Moreover, the comparison results show that the performance provided by the tasks mentioned above (CoLA, SST-2, RTE, MNLI and QNLI) does not depend on the used text summarization dataset, which means the best performing models are always the same for the three datasets(DUC’2002, DUC’2003 and DUC’2004).
Second, the experiments are conducted to address the following research question: (2) Does BERT multi-task fine-tuning improve the performance of extractive multi-document summarization task?
The last block of Table 4 reports the obtained results by the fine-tuned BERT models using multi-task learning:
4.5. Comparison evaluation with state-of-the-art methods
To validate the robustness of the proposed method, we compare its performance with existing state-of-the-art methods for generic multi-document summarization task [62], using the evaluation measures R-1, R-2, R-4 and R-L. For this purpose, we used the repository developed by [63], which contains a set of summaries generated by several systems for DUC’2004 dataset. The generated summaries are published in GitHub 8 while the source codes of most of these systems are available in Sumy repository. 9 The obtained results are shown in Table 5. The first block of the table summarises the results of the systems used for the comparison, while the second block shows the best results obtained by our method.
Systems performance on DUC’2004, using state-of-the-art methods and the proposed method. The highest performance for each of the group of methods is printed in boldface.
For denoting statistical significance results, the superscript numbers indicate significant improvement (p value <0.05) over the task used for BERT Fine-Tuning that has the same superscript number. The interval
‡The results of models are taken from their original articles.
On the one hand, the first set of analysis aims to highlight the importance of documents representations for extractive text summarization task. For this purpose, we compare our method with two state-of-the-art methods, including the Centroid_BOW method that uses bag-of-words representations and the Centroid_WordEmbedding method that exploits the capabilities of word embedding models. The obtained results show that our method using the fine-tuned BERT model, considered as a sentence embedding model, has significantly outperformed the centroid method based on BOW representations for most evaluation measures. Furthermore, the overall comparison results demonstrate that our method achieved far better performances than the Centroid_WordEmbedding method in terms of R-1 measure. For instance, with the
On the other hand, the second set of evaluations is conducted to compare our method with several baselines and state-of-the-art methods for extractive generic multi-document summarization. For the R-1 evaluation measure, our method using
5. Discussion
The pre-trained BERT model has shown outstanding performances in several natural language processing applications. However, the obtained results show that its application in extractive text summarization without fine-tuning does not achieve good results and lags significantly behind the performance of the most fine-tuned BERT models. The latter can be explained by the fact that BERT model is trained on a masked language model where the output vectors are grounded to tokens instead of sentences, while sentence representation is the cornerstone of our unsupervised extractive multi-document summarization methods. Therefore, to improve the performance of sentence representation learning, we fine-tuned BERT model on intermediate tasks before applying it in our method using both single-task and multi-task fine-tuning. Hence, we used GLUE benchmark tasks, including the natural language inference (NLI) tasks (RTE, MNLI and QNLI, the similarity and paraphrasing tasks (STS-B, MRPC and QQP), and the single sentence classification tasks (CoLA and SST-2).
The overall obtained results show that fine-tuning BERT model on NLI tasks (RTE, MNLI and QNLI) has achieved the best performances, where fine-tuning BERT on the RTE dataset outperforms all others. Moreover, fine-tuning BERT on MNLI or QNLI has obtained comparable performance. These results might be due to the fact that the latter are among high-quality labelled corpus, designed for textual entailment tasks, which might constitute a class of problems relevant to text summarization task [72,73]. Moreover, Fine-tuning BERT on single sentence classification tasks (CoLA and SST-2) has shown promising results. Even though CoLA task is a challenging task with much smaller in-domain data than other tasks, it was able to transfer useful knowledge to the text summarization task. The latter can be explained by the fact that CoLA aims to predict the linguistic acceptability of a sentence, which may require a model that captures syntactic and semantic information of the sentences. Although fine-tuning BERT on sentence similarity task (STS-B) and paraphrasing tasks (QQP and MRPC) has outperformed the pre-trained model, these tasks have led to poor performance.
Besides, it is important to highlight that our method has achieved better performance when BERT is fine-tuned on tasks that only have small amounts of training datasets than those of large training datasets. Even if the tasks belong to the same type, for example, the two NLI tasks RTE vs MNLI, and the single sentence classification tasks CoLA vs SST-2. In addition, we also note that the performance of
Furthermore, the results show that BERT multi-task fine-tuning has outperformed models that are fine-tuned on single-tasks for text summarization. Hence, multi-task learning is an important paradigm to learn universal sentence representations. This stems from the fact that the latter improve generalisation performance by taking advantage of the training signals of related tasks [13,47,48].
Finally, the overall comparison results show that our method has achieved significant improvements over several baselines and state-of-the-art methods. This finding proves the effectiveness of fine-tuning BERT model on intermediate tasks that improves sentence representation learning.
6. Conclusion
In this article, we introduced an unsupervised method for extractive multi-document summarization that consists of two main tasks: BERT fine-tuning and multi-document summarization. First, we fine-tune the pre-trained BERT model using two different approaches, namely BERT single-task fine-tuning and BERT multi-task fine-tuning. Second, we use the obtained fine-tuned BERT models to represent the sentences of the input documents. Then, based on the latter representations, we assign a score to each sentence in the cluster of documents by combining three scores including sentence content relevance, sentence novelty and sentence position where the top-ranked sentences are iteratively selected to form the final summary.
To assess the effectiveness of the proposed multi-document summarization method, we have performed an extensive experimental analysis. Indeed, several experiments have been conducted on the standard DUC benchmark datasets (2002–2004) to investigate the GLUE tasks that allow transferring useful knowledge to multi-document summarization task. The overall obtained results showed that fine-tuning BERT model on CoLA, SST-2, RTE, MNLI and QNLI tasks had achieved the best performance and improved the results significantly. In addition, according to ROUGE evaluation measures, the results demonstrate that fine-tuning BERT model using multi-task learning has improved the performance of the extractive multi-document summarization task. Moreover, the overall comparison results show that our method obtained promising results; it has significantly outperformed several baseline methods and achieved far better performances than two state-of-the-art deep learning–based methods (GBN-MDS, PG-MMR) for most evaluation measures.
The use of fine-tune BERT models in extractive multi-document summarization has shown to be effective. Hence, in future work, we plan to investigate their efficiency on other summarization tasks such as query-focused summarization. In addition, we plan to explore the potential of BERT model for language generation.
Footnotes
Declaration of conflicting interests
The author(s) declared no potential conflicts of interest with respect to the research, authorship and/or publication of this article.
Funding
The author(s) received no financial support for the research, authorship, and/or publication of this article.
