Ad-Hoc Document Retrieval Using Weak-Supervision with BERT and GPT2

Total Page:16

File Type:pdf, Size:1020Kb

Ad-Hoc Document Retrieval Using Weak-Supervision with BERT and GPT2 Ad-hoc Document Retrieval using Weak-Supervision with BERT and GPT2 Yosi Mass Haggai Roitman∗ IBM Research eBay Research Haifa University, Mount Carmel, Haifa, Netanya, Israel HA 31905, Israel [email protected] [email protected] Abstract stracts can take the role of questions and answers of FAQs, respectively. We describe a weakly-supervised method for Whenever a document is missing a title, we con- training deep learning models for the task of sider its first sentence as its augmented title. In ad-hoc document retrieval. Our method is based on generative and discriminative mod- a similar way, whenever a document is missing els that are trained using weak-supervision an abstract, we consider the first 512 words of its based solely on the documents in the cor- content as the abstract. pus. We present an end-to-end retrieval sys- The three fields are used for retrieving candidate tem that starts with traditional information re- documents. Inspired by (Mass et al., 2020), the trieval methods, followed by two deep learning title and abstract fields are further used as a weak- re-rankers. We evaluate our method on three supervision data source for training two indepen- different datasets: a COVID-19 related scien- dent BERT (Devlin et al., 2019) models, that are tific literature dataset and two news datasets. We show that our method outperforms state-of- then used to re-rank those candidates documents. the-art methods; this without the need for the The first model matches user queries to docu- expensive process of manually labeling data. ments’ abstracts. Here we use the title-to-abstract associations to fine-tune a BERT model to semanti- 1 Introduction cally match queries to abstracts. The second model matches user queries to titles. Here our assumption The ad-hoc retrieval task has been extensively is that by generating title paraphrases, we can train studied by the Information Retrieval (IR) com- a model to match user queries to titles. To this end, munity. Traditional IR models evaluate ad-hoc we use GPT2 (Radford et al., 2018) to generate title queries against documents mainly on a syntac- paraphrases, which are then utilized for fine-tuning tic (exact) word-matching basis (Manning et al., the second BERT model. 2008). Recent years advances in Deep Learning While our work is closely related to (Mass et al., (DL) methods have lead to further improvement in 2020), with the lack of human-curated questions IR tasks, and among others, in ad-hoc document (such as in FAQs), we still need to resort to title retrieval (Guo et al., 2019). DL methods add a se- paraphrases as (noisy) pseudo-questions and trans- mantic dimension to IR methods. However, such fer (Mass et al., 2020)’s method to the more general methods usually require large amounts of labeled task of ad-hoc document retrieval. Moreover, com- data for model training. pared to FAQs that are relatively short, the current In this work, we describe a novel weakly- task deals with documents that can be quite long. supervised method for training DL methods for Thus, in current paper we use three fields (title, ab- ad-hoc document retrieval. Motivated by the re- stract, content) and present a strong IR base line cent work of (Mass et al., 2020) on Frequently instead of only two fields and a simple IR baseline Asked Questions (FAQ) retrieval, we assume that used in (Mass et al., 2020) documents have at least three fields, namely title, As a proof of concept, we evaluate our method abstract and content. Such documents are actually on three benchmarks: TREC-COVID - a scien- quite common nowadays in the scientific and news tific literature dataset on COVID-19 topics; and domains. Our main hypothesis is that: titles and ab- TREC’s newswire corpora: Associated Press (AP) *∗ Work done while affiliated with IBM. and Wall Street Journal (WSJ). By combining the 4191 Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing, pages 4191–4197, November 16–20, 2020. c 2020 Association for Computational Linguistics two weakly-supervised BERT models with an ex- a joint attention-based representation for pairs of isting strong IR baseline, we demonstrate that the (query, abstarct) and (query, title) while in (Chang former can help to elevate the performance of the et al., 2020) they learn a separate representation for latter. Our approach further outperforms state-of- queries and passages. the-art methods on these benchmarks. 3 Method 2 Related Work Inspired by (Mass et al., 2020), we consider the ad- With the lack of training data, several weakly- hoc document retrieval problem as an instance of supervised alternatives have been explored so far FAQ retrieval, where a document’s title represents for the task at hand. (Dehghani et al., 2017b,a) and the question and its abstract the answer. (Nie et al., 2018) have utilized rankings produced Our proposed retrieval approach allows to en- by BM25 model as training samples. (MacAvaney hance existing state-of-the-art ad-hoc retrieval et al., 2019) have used pseudo query-document methods with weakly-supervised neural models pairs that already exhibit relevance (e.g., newswire that are completely trained from the documents headline-content pairs). (Frej et al., 2019) have collection itself without the need to supply man- utilized Wikipedia’s internal linkage to define au- ual relevance labels. Following the common ap- tomated queried topics. (Zhang et al., 2020) have proach (Guo et al., 2019), these neural-models used anchor texts and their linked web pages as are utilized for re-ranking candidate documents query-document pairs. retrieved by a given IR baseline. Our work is different from all those works as we In what follows, the initial candidate documents train a model to generate title paraphrases that are retrieval uses pure IR similarities and relevance used to enable query-to-title (question) matching models (Section 3.1). The re-ranking step exploits and not only query-to-abstract (answer) matching. two independent weakly-supervised BERT models, (Ma et al., 2020) have proposed a zero-shot re- namely: BERT-Q-a (Section 3.2) for matching trieval approach using synthetic query generation queries to abstracts and BERT-Q-t (Section 3.2) by training a generative model on a different Com- for matching queries to titles. munity QA data. Our work differs from (Ma et al., The final re-ranking is obtained by combining 2020) in three main aspects. First, (Ma et al., 2020) the outcome of the baseline IR method and the two focuses on QA, where answers are very short, while BERT-based re-rankers using an unsupervised late- we generate title paraphrases from full abstracts. fusion step (Section 3.4). The components of our Second, we train a model to generate title para- approach are described in the rest of this section. phrases which are used to enable not only query-to- abstract (answer) matching, but also query-to-title 3.1 Initial retrieval (question) matching. Third, (Ma et al., 2020) fil- ters the input QA pairs that are used to train the We first obtain for each query a reasonable pool generative model by taking only pairs that were of candidate documents to be re-ranked using our voted by at least one-user on those Community QA weakly-supervised models. To this end we retrieve (CQA) sites. We do not have such voting so we several ranked lists from an Apache Lucene1 index use a smart filtering on the output data (namely using various state-of-the-art IR similarities. that on the generated title-paraphrases) as described in are available in Lucene. The various retrieved lists Section 3.3. are then combined to generate a single pool of The work in (Chang et al., 2020) suggests an top-k candidates for re-ranking by employing the efficient neural method for initial retrieval of candi- PoolRank (Roitman, 2018) fusion method. We dates. Their method uses a two-tower architecture refer to this IR pipeline as IR-Base. which learns a different representation for passages The IR similarities and the PoolRank method and for queries. While their method can be used have few free-parameters that are tuned so to opti- as an initial retrieval (instead of our IR method), mize Mean Average Precision (MAP@1000). De- the authors of (Chang et al., 2020) still require tails are given in the experimental setup (Sec- an additional re-ranking step. Thus it does not re- tion 4.2) below. place our two weakly-supervised BERT re-ranking models. Moreover, our two BERT models learn 1https://lucene.apache.org/ 4192 3.2 BERT-Q-a At run time, given a user query Q, BERT-Q-t re- k We use pairs of title-abstract (t,a) of documents in ranks the top- candidate documents by matching Q t the collection as a weak-supervision data source to titles ( ) only. for fine-tuning a pre-trained BERT model which is 3.4 Enhanced ad-hoc retrieval using Fusion then used to match user queries to abstracts. To enhance ad-hoc retrieval quality, we now pro- Similar to (Mass et al., 2020), we fine-tune the pose to combine the two weakly-supervised fine- BERT model (denoted BERT-Q-a) using a triplet tuned BERT models with the baseline IR method network (Hoffer and Ailon, 2015). This network is (IR-Base, see again Section 3.1). To this end, fol- adopted for BERT fine-tuning (Mass et al., 2019) lowing (Roitman, 2018), we utilize the Two-Step using triplets (t; a; a0), where (t; a) constitutes a PoolRank (denoted TSPR) unsupervised fusion document title and its abstract.
Recommended publications
  • Generating Multi-Agent Trajectories Using Programmatic Weak
    Published as a conference paper at ICLR 2019 GENERATING MULTI-AGENT TRAJECTORIES USING PROGRAMMATIC WEAK SUPERVISION Eric Zhan Stephan Zheng∗ Caltech Salesforce [email protected] [email protected] Yisong Yue Long Sha & Patrick Lucey Caltech STATS [email protected] flsha,[email protected] ABSTRACT We study the problem of training sequential generative models for capturing co- ordinated multi-agent trajectory behavior, such as offensive basketball gameplay. When modeling such settings, it is often beneficial to design hierarchical mod- els that can capture long-term coordination using intermediate variables. Further- more, these intermediate variables should capture interesting high-level behavioral semantics in an interpretable and manipulatable way. We present a hierarchical framework that can effectively learn such sequential generative models. Our ap- proach is inspired by recent work on leveraging programmatically produced weak labels, which we extend to the spatiotemporal regime. In addition to synthetic settings, we show how to instantiate our framework to effectively model complex interactions between basketball players and generate realistic multi-agent trajec- tories of basketball gameplay over long time periods. We validate our approach using both quantitative and qualitative evaluations, including a user study compar- ison conducted with professional sports analysts.1 1 INTRODUCTION The ongoing explosion of recorded tracking data is enabling the study of fine-grained behavior in many domains: sports (Miller et al., 2014; Yue et al., 2014; Zheng et al., 2016; Le et al., 2017), video games (Ross et al., 2011), video & motion capture (Suwajanakorn et al., 2017; Taylor et al., 2017; Xue et al., 2016), navigation & driving (Ziebart et al., 2009; Zhang & Cho, 2017; Li et al., 2017), lab- oratory animal behaviors (Johnson et al., 2016; Eyjolfsdottir et al., 2017), and tele-operated robotics (Abbeel & Ng, 2004; Lin et al., 2006).
    [Show full text]
  • Constrained Labeling for Weakly Supervised Learning
    Constrained Labeling for Weakly Supervised Learning Chidubem Arachie1 Bert Huang2 1Department of Computer Science, Virginia Tech, Blacksburg, Virginia, USA 2Department of Computer Science, Data Intensive Studies Center, Tufts University, Medford, Massachusetts, USA Abstract sion. Weakly supervised learning involves training models using noisy labels. Using multiple sources or forms of weak supervision is common, as it provides diverse information Curation of large fully supervised datasets has be- to the model. However, each source of weak supervision has come one of the major roadblocks for machine its own bias that can be transmitted to the model. Different learning. Weak supervision provides an alterna- weak supervision signals can also conflict, overlap, or—in tive to supervised learning by training with cheap, the worst case—make dependent errors. Thus, a naive com- noisy, and possibly correlated labeling functions bination of these weak signals would hurt the quality of from varying sources. The key challenge in weakly a learned model. The key problem then is how to reliably supervised learning is combining the different combine various sources of weak signals to train an accurate weak supervision signals while navigating mislead- model. ing correlations in their errors. In this paper, we propose a simple data-free approach for combin- To solve this problem, we propose constrained label learn- ing weak supervision signals by defining a con- ing (CLL), a method that processes various weak supervi- strained space for the possible labels of the weak sion signals and combines them to produce high-quality signals and training with a random labeling within training labels. The idea behind CLL is that, given the weak this constrained space.
    [Show full text]
  • Multi-Task Weak Supervision Enables Automated Abnormality Localization in Whole-Body FDG-PET/CT
    Multi-task weak supervision enables automated abnormality localization in whole-body FDG-PET/CT Sabri Eyuboglu∗1, Geoffrey Angus∗1, Bhavik N. Patel2, Anuj Pareek2, Guido Davidzon2, Jared Dunnmon,∗∗1 Matthew P. Lungren ∗∗2 1 Department of Computer Science, Stanford University, Stanford, CA 94305, USA 2 Department of Radiology, Stanford University, Stanford, CA 94305, USA ∗ ∗∗ Equal contribution; Computational decision support systems could provide clinical value in whole-body FDG- PET/CT workflows. However, the lack of labeled data combined with the sheer size of each scan make it challenging to apply existing supervised machine learning systems. Leveraging recent advancements in natural language processing, we develop a weak supervision frame- work that extracts imperfect, yet highly granular regional abnormality labels from the radi- ologist reports associated with each scan. Our framework automatically labels each region in a custom ontology of anatomical regions, providing a structured profile of the pathologies in each scan. We design an attention-based, multi-task CNN architecture that can be trained using these generated labels to localize abnormalities in whole-body scans. We show that our multi-task representation is critical for strong performance on rare abnormalities for which we have little data and for accurately predicting mortality from imaging data alone (AUROC 80%), a critical challenge for palliative care teams. 1 Introduction The availability of large, labeled datasets has fueled recent progress in machine learning for med- ical imaging. Breakthrough studies in chest radiograph diagnosis, skin and mammographic lesion classification, and diabetic retinopathy detection have relied on hundreds of thousands of labeled training examples [1–4].
    [Show full text]
  • Visual Learning with Weak Supervision
    Visual Learning with Weak Supervision Safa Cicek Committee: Prof. Stefano Soatto, Chair Prof. Lieven Vandenberghe Prof. Paulo Tabuada Prof. Guy Van den Broeck PHD Defense January 2021 UCLA ELECTRICAL AND COMPUTER ENGINEERING Table of Contents Introduction SaaS: Speed as a Supervisor for Semi-supervised Learning [1] Input and Weight Space Smoothing for Semi-supervised Learning [2] Unsupervised Domain Adaptation via Regularized Conditional Alignment [3] Disentangled Image Generation for Unsupervised Domain Adaptation [4] Spatial Class Distribution Shift in Unsupervised Domain Adaptation [5] Learning Topology from Synthetic Data for Unsupervised Depth Completion [6] Targeted Adversarial Perturbations for Monocular Depth Prediction [7] Concluding Remarks [1] Cicek, Safa, Alhussein Fawzi, and Stefano Soatto. Saas: Speed as a supervisor for semi-supervised learning. Proceedings of the European Conference on Computer Vision (ECCV). 2018. [2] Cicek, Safa, and Stefano Soatto. Input and Weight Space Smoothing for Semi-supervised Learning. Proceedings of the IEEE International Conference on Computer Vision (ICCV) Workshops. 2019. [3] Cicek, Safa, and Stefano Soatto. Unsupervised domain adaptation via regularized conditional alignment. Proceedings of the IEEE International Conference on Computer Vision (ICCV). 2019. [4] Cicek, Safa, Zhaowen Wang, Hailin Jin, Stefano Soatto, Generative Feature Disentangling for Unsupervised Domain Adaptation. Proceedings of the European Conference on Computer Vision (ECCV) Workshops. (2020). [5] Cicek, Safa, Ning Xu, Zhaowen Wang, Hailin Jin, Stefano Soatto, Spatial Class Distribution Shift in Unsupervised Domain Adaptation. Asian Conference on Computer Vision (ACCV). 2020. [6] Wong Alex, Safa Cicek, Stefano Soatto, Learning Topology from Synthetic Data for Unsupervised Depth Completion, IEEE Robotics and Automation Letters (RAL). 2021. [7] Wong Alex, Safa Cicek, Stefano Soatto, Targeted Adversarial Perturbations for Monocular Depth Prediction.
    [Show full text]
  • Natural Language Processing and Weak Supervision
    Natural language processing and weak supervision L´eonBottou COS 424 { 4/27/2010 Introduction Natural language processing \from scratch" { Natural language processing systems are heavily engineered. { How much engineering can we avoid by using more data ? { Work by Ronan Collobert, Jason Weston, and the NEC team. Summary { Natural language processing { Embeddings and models { Lots of unlabeled data { Task dependent hacks 2/43 COS 424 { 4/27/2010 I. Natural language processing The Goal We want to have a conversation with our computer . still a long way before HAL 9000 . Convert a piece of English into a computer-friendly data structure How to measure if the computer \understands" something? 4/43 COS 424 { 4/27/2010 Natural Language Processing Tasks Intermediate steps to reach the goal? Part-Of-Speech Tagging (POS): syntactic roles (noun, adverb...) Chunking (CHUNK): syntactic constituents (noun phrase, verb phrase...) Name Entity Recognition (NER): person/company/location... Semantic Role Labeling (SRL): semantic role [John]ARG0 [ate]REL [the apple]ARG1 [in the garden]ARGM LOC − 5/43 COS 424 { 4/27/2010 NLP Benchmarks Datasets: ? POS, CHUNK, SRL: WSJ( up to 1M labeled words) ? NER: Reuters( 200K labeled≈ words) ≈ System Accuracy System F1 Shen, 2007 97.33% Shen, 2005 95.23% Toutanova, 2003 97.24% Sha, 2003 94.29% Gimenez, 2004 97.16% Kudoh, 2001 93.91% (a) POS: As in (Toutanova, 2003) (b) CHUNK: CoNLL 2000 System F1 System F1 Ando, 2005 89.31% Koomen, 2005 77.92% Florian, 2003 88.76% Pradhan, 2005 77.30% Kudoh, 2001 88.31% Haghighi, 2005
    [Show full text]
  • Towards a Perceptually Relevant Control Space
    Music sound synthesis using machine learning: Towards a perceptually relevant control space Fanny ROCHE PhD Defense 29 September 2020 Supervisor: Laurent GIRIN Co-supervisors: Thomas HUEBER Co-supervisors: Maëva GARNIER Co-supervisors: Samuel LIMIER Fanny ROCHE - PhD Defense - 29 September 2020 1 / 48 Context and Objectives Context and Objectives Fanny ROCHE - PhD Defense - 29 September 2020 2 / 48 Context and Objectives Context and Objectives Fanny ROCHE - PhD Defense - 29 September 2020 2 / 48 Context and Objectives Context and Objectives Fanny ROCHE - PhD Defense - 29 September 2020 2 / 48 Processed recordings ! e.g. sampling, wavetable, granular, ... + computationally efficient − very memory consuming Spectral modeling ! e.g. additive, subtractive, ... + close to human sound perception − numerous & very specific Physical modeling ! e.g. solving wave equations, modal synthesis, ... + physically meaningful controls − very specific sounds Context and Objectives Context and Objectives Music sound synthesis Abstract algorithms ! e.g. FM, waveshaping, ... + rich sounds − complex parameters Fanny ROCHE - PhD Defense - 29 September 2020 3 / 48 Spectral modeling ! e.g. additive, subtractive, ... + close to human sound perception − numerous & very specific Physical modeling ! e.g. solving wave equations, modal synthesis, ... + physically meaningful controls − very specific sounds Context and Objectives Context and Objectives Music sound synthesis Abstract algorithms ! e.g. FM, waveshaping, ... + rich sounds − complex parameters Processed recordings ! e.g. sampling, wavetable, granular, ... + computationally efficient − very memory consuming Fanny ROCHE - PhD Defense - 29 September 2020 3 / 48 Physical modeling ! e.g. solving wave equations, modal synthesis, ... + physically meaningful controls − very specific sounds Context and Objectives Context and Objectives Music sound synthesis Abstract algorithms ! e.g. FM, waveshaping, ... + rich sounds − complex parameters Processed recordings ! e.g.
    [Show full text]
  • Self-Supervised Collaborative Learning for Medical Image Synthesis
    The Thirty-Fourth AAAI Conference on Artificial Intelligence (AAAI-20) Auto-GAN: Self-Supervised Collaborative Learning for Medical Image Synthesis Bing Cao,1 Han Zhang,2∗ Nannan Wang,1∗ Xinbo Gao,1 Dinggang Shen2∗ 1State Key Laboratory of Integrated Services Networks, Xidian University, Xi’an, China 2Department of Radiology and BRIC, University of North Carolina at Chapel Hill, Chapel Hill, USA Abstract tors desire as many different modalities as possible to make a joint and more accurate decision, certain limitations (e.g., In various clinical scenarios, medical image is crucial in dis- restricted medical conditions, inadequate scanning time, and ease diagnosis and treatment. Different modalities of medical images provide complementary information and jointly helps cost/spending/resource control) could result in sacrificing doctors to make accurate clinical decision. However, due to some imaging modalities to have the most important one clinical and practical restrictions, certain imaging modalities done. For examples, low-resolution (e.g., thick-sliced) imag- may be unavailable nor complete. To impute missing data ing modalities are always combined with another single with adequate clinical accuracy, here we propose a frame- high-resolution imaging modality to save scanning time and work called self-supervised collaborative learning to syn- the low-resolution images can be misleading due to its par- thesize missing modality for medical images. The proposed tial coverage. CT is notorious for poor soft-tissue contrast method comprehensively utilize all available information cor- and radiation exposure, but can better quantify tissue’s phys- related to the target modality from multi-source-modality im- ical density. MRI is non-invasive and safer with better con- ages to generate any missing modality in a single model.
    [Show full text]
  • Weakly Supervised Text Classification with Naturally Annotated Resource
    Automated Knowledge Base Construction (2021) Conference paper NatCat: Weakly Supervised Text Classification with Naturally Annotated Resources Zewei Chu [email protected] University of Chicago, Chicago, IL 60637, USA Karl Stratos [email protected] Rutgers University, Piscataway, NJ 08854, USA Kevin Gimpel [email protected] Toyota Technological Institute at Chicago, Chicago, IL 60637, USA Abstract We describe NatCat, a large-scale resource for text classification constructed from three data sources: Wikipedia, Stack Exchange, and Reddit. NatCat consists of document- category pairs derived from manual curation that occurs naturally within online communities. To demonstrate its usefulness, we build general purpose text classifiers by training on NatCat and evaluate them on a suite of 11 text classification tasks (CatEval), reporting large improvements compared to prior work. We benchmark different modeling choices and resource combinations and show how tasks benefit from particular NatCat data sources.1 1. Introduction Websites with community contributed content contain ample knowledge for natural lan- guage processing. In this paper, we seek to improve text classification by leveraging this knowledge in the form of texts paired with natural category annotations. In particular, we present NatCat, a large-scale resource constructed automatically from three data sources: Wikipedia, Stack Exchange,2 and Reddit.3 With rich knowledge about text and categories, we show that NatCat is a useful resource for text classification. To demonstrate the usefulness of NatCat, we use it to train models that compute a score for any document-category pair. As a result, we get weakly supervised (i.e., \dataless"; Chang et al., 2008) text classifiers that can be used off-the-shelf to produce interpretable and relevant topics for any document.
    [Show full text]
  • Weakly Supervised Short Text Categorization Using World Knowledge
    Weakly Supervised Short Text Categorization Using World Knowledge Rima T¨urker1;2[0000−0001−8332−7241], Lei Zhang1[0000−0001−8184−9353], Mehwish Alam1;2[0000−0002−7867−6612], and Harald Sack1;2[0000−0001−7069−9804] 1 FIZ Karlsruhe { Leibniz Institute for Information Infrastructure, Germany 2 Karlsruhe Institute of Technology, Institute AIFB, Germany [email protected] [email protected] Abstract. Short text categorization is an important task in many NLP applications, such as sentiment analysis, news feed categorization, etc. Due to the sparsity and shortness of the text, many traditional classifi- cation models perform poorly if they are directly applied to short text. Moreover, supervised approaches require large amounts of manually la- beled data, which is a costly, labor intensive, and time-consuming task. This paper proposes a weakly supervised short text categorization ap- proach, which does not require any manually labeled data. The proposed model consists of two main modules: (1) a data labeling module, which leverages an external Knowledge Base (KB) to compute probabilistic labels for a given unlabeled training data set, and (2) a classification model based on a Wide & Deep learning approach. The effectiveness of the proposed method is validated via evaluation on multiple datasets. The experimental results show that the proposed approach outperforms unsupervised state-of-the-art classification approaches and achieves com- parable performance to supervised approaches. Keywords: Short Text Categorization · Weakly Supervised Short Text Categorization · Wide & Deep Model. 1 Introduction Due to rapid growth of the Web content, short text data such as search snippets, news feeds, short messages, etc.
    [Show full text]
  • Strength from Weakness: Fast Learning Using Weak Supervision
    Strength from Weakness: Fast Learning Using Weak Supervision Joshua Robinson 1 Stefanie Jegelka 1 Suvrit Sra 1 Abstract plex model on a related, large data task, and to then uses We study generalization properties of weakly su- the learned features for finetuning on the small-data target pervised learning, that is, learning where only a task (Girshick et al., 2014; Donahue et al., 2014; Zeiler & few “strong” labels (the actual target for predic- Fergus, 2014; Sun et al., 2017). Numerous approaches to tion) are present but many more “weak” labels are weakly supervised learning have succeeded in a variety of available. In particular, we show that pretraining tasks; beyond computer vision (Oquab et al., 2015; Durand using weak labels and finetuning using strong can et al., 2017; Carreira & Zisserman, 2017; Fries et al., 2019). accelerate the learning rate for the strong task to Examples include clinical text classification (Wang et al., 2019), sentiment analysis (Medlock & Briscoe, 2007), so- the fast rate of (1/n), where n is the number of strongly labeledO data points. This acceleration cial media content tagging (Mahajan et al., 2018) and many can happen even if, by itself, the strongly labeled others. Weak supervision is also closely related to unsu- pervised learning methods such as complementary and con- data admits only the slower (1/√n) rate. The acceleration depends continuouslyO on the number trastive learning (Xu et al., 2019; Chen & Batmanghelich, of weak labels available, and on the relation be- 2019; Arora et al., 2019), and particularly to self-supervised tween the two tasks. Our theoretical results are learning (Doersch et al., 2015), where feature maps learned reflected empirically across a range of tasks and via supervised training on artificially constructed tasks have illustrate how weak labels speed up learning on been found to even outperform ImageNet learned features on the strong task.
    [Show full text]
  • Snuba: Automating Weak Supervision to Label Training Data
    Snuba: Automating Weak Supervision to Label Training Data Paroma Varma Christopher Re´ Stanford University Stanford University [email protected] [email protected] ABSTRACT benign, malignant if area > 210.8: As deep learning models are applied to increasingly diverse return False Label if area < 150: Aggregator problems, a key bottleneck is gathering enough high-quality return Abstain training labels tailored to each task. Users therefore turn Labeled Data 25% benign, 75%benign, ... ?, ?, ?, ? if perim > 120: to weak supervision, relying on imperfect sources of labels return True like pattern matching and user-defined heuristics. Unfor- if perim > 80 Terminate? return Abstain tunately, users have to design these sources for each task. Heuristic Generation Training Labels This process can be time consuming and expensive: domain Unlabeled Data Snuba experts often perform repetitive steps like guessing optimal numerical thresholds and developing informative text pat- Figure 1: Snuba uses a small labeled and a large unlabeled terns. To address these challenges, we present Snuba,a dataset to iteratively generate heuristics. It uses existing la- system to automatically generate heuristics using a small bel aggregators to assign training labels to the large dataset. labeled dataset to assign training labels to a large, unla- beled dataset in the weak supervision setting. Snuba gen- pervision, or methods that can assign noisy training labels erates heuristics that each labels the subset of the data it to unlabeled data, like crowdsourcing [9, 22, 60], distant su- is accurate for, and iteratively repeats this process until the pervision [8,32], and user-defined heuristics [38,39,50]. Over heuristics together label a large portion of the unlabeled the past few years, we have been part of the broader effort data.
    [Show full text]
  • Multi-Modal Weakly Supervised Dense Event Captioning
    Watch, Listen and Tell: Multi-modal Weakly Supervised Dense Event Captioning Tanzila Rahman1,2 Bicheng Xu1,2 Leonid Sigal1,2,3 1University of British Columbia 2Vector Institute for AI 3Canada CIFAR AI Chair {trahman8, bichengx, lsigal}@cs.ubc.ca Abstract Input Video with Audio Track Dense Event Captions A group of friends sit Multi-modal learning, particularly among imaging and together in a living room linguistic modalities, has made amazing strides in many talking. high-level fundamental visual understanding problems, .. ranging from language grounding to dense event caption- One of the friends plays . a song on his guitar and ing. However, much of the research has been limited to the others listen. approaches that either do not take audio corresponding to .. Time video into account at all, or those that model the audio- . One friend falls back visual correlations in service of sound or sound source lo- into the chair in . amazement after the calization. In this paper, we present the evidence, that audio song . signals can carry surprising amount of information when .. it comes to high-level visual-lingual tasks. Specifically, The two friends congratulate the we focus on the problem of weakly-supervised dense event man for his guitar performance and captioning in videos and show that audio on its own can shake hands. nearly rival performance of a state-of-the-art visual model and, combined with video, can improve on the state-of-the- Figure 1: Multi-modal Dense Event Captioning. Illus- art performance. Extensive experiments on the ActivityNet tration of our problem definition, where we use both audio Captions dataset show that our proposed multi-modal ap- features and visual information to generate the dense cap- proach outperforms state-of-the-art unimodal methods, as tions for a video in a weakly supervised manner.
    [Show full text]