Papers.
Research connected to its authors, projects, companies, talks, events, and the rest of the graph.
Add a paper ↗Neural representations for modeling variation in speech
Journal of Phonetics · DOI 10.1016/j.wocn.2022.101137 · 29 citations · Source: openalex+first-party-career-authorityMark Liberman, Martijn Bartelds, Wietse de Vries, Faraz Sanal, Caitlin Richter, Martijn Wieling · 6 authors totalAccurate Clinical and Biomedical Named Entity Recognition at Scale
Software Impacts · DOI 10.1016/j.simpa.2022.100373 · 40 citations · Source: openalexWe introduce an agile, production-grade clinical and biomedical Named entity recognition (NER) algorithm based on a modified BiLSTM-CNN-Char DL architecture built on top of Apache Spark. Our NER implementation establishes new state-of-the-art accuracy on 7 of 8 well-known biomedical NER benchmarks and 3 clinical concept extraction challenges: 2010 i2b2/VA clinical concept extraction, 2014 n2c2 de-identification, and 2018 n2c2 medication extraction. Moreover, clinical NER models trained using this implementation outperform the accuracy of commercial entity extraction solutions, AWS Medical Comprehend and Google Cloud Healthcare API by a large margin (8.9% and 6.7% respectively), without using memory-intensive language models.
David Talby, Veysel Kocaman · 2 authors totalConnecting the dots in clinical document understanding with Relation Extraction at scale
Software Impacts · DOI 10.1016/j.simpa.2022.100294 · 0 citations · Source: openalexWe present a text mining framework based on top of the Spark NLP library — comprising of Named Entity Recognition (NER) and Relation Extraction (RE) models, which expands on previous work in three main ways. First, we release new RE model architectures that obtain state-of-the-art F1 scores on 5 out of 7 benchmark datasets. Second, we introduce a modular approach to train and stack multiple models in a single nlp pipeline in a production grade library with little coding. Third, we apply these models in practical applications including knowledge graph generation, prescription parsing, and robust ontology mapping.
David Talby, Hasham Ul Haq, Veysel Kocaman · 3 authors totalNatural speech markers of Alzheimer's disease co-pathology in Lewy body dementias
Parkinsonism & Related Disorders · DOI 10.1016/j.parkreldis.2022.07.023 · 14 citations · Source: openalex+first-party-career-authorityMark Liberman, Sanjana Shellikeri, Sunghye Cho, Katheryn A Q Cousins, Erica Howard, Yvonne Balganorth, Daniel Weintraub, Meredith Spindler · 15 authors totalDecentralizing components of electronic markets to prevent gatekeeping and manipulation
Electronic Commerce Research and Applications · DOI 10.1016/j.elerap.2022.101220 · 6 citations · Source: openalex+orcid+dblp-identityJohan Pouwelse, Martijn de Vos, Georgy Ishmaev · 3 authors totalDimensions of Speech and Language Disturbance in Psychosis and Computational Linguistic Markers
Biological Psychiatry · DOI 10.1016/j.biopsych.2022.02.144 · 2 citations · Source: openalex+first-party-career-authorityMark Liberman, Sunny X. Tang, Katrin Hänsel, Yan Cong, Sarah Berretta, Sunghye Cho, Amir Hossein Nikzad, Aarush Mehta · 10 authors totalCharacterizing Twitter Influencers in Radiation Oncology
Advances in Radiation Oncology · DOI 10.1016/j.adro.2022.100919 · 9 citations · Source: openalex+first-party-career-authorityMarc Smith, Luca Valle, Fang‐I Chu, Marc A. Smith, Chenyang Wang, Percy Lee, Drew Moghanaki, Fumiko Chino · 9 authors totalOutsourcing as a cooperative game between the CEO and labor: theory and evidence
Review of Quantitative Finance and Accounting · DOI 10.1007/s11156-022-01071-x · 0 citations · Source: semantic-scholarJose Plehn, J. Choi, Ming Ju, J. Plehn-Dujowich, Xiaotian Zhang · 5 authors totalOn the optimization landscape of tensor decompositions.
Math. Program. · DOI 10.1007/s10107-020-01579-x · Source: dblp+stanford-authorityTengyu Ma, Rong Ge 0001, Tengyu Ma 0001 · 3 authors totalLane-Level Route Planning for Autonomous Vehicles
ISRR 2022 (Springer Proceedings in Advanced Robotics) · DOI 10.1007/978-3-031-21090-7_19 · 3 citations · Source: openalexJur van den Berg, Mitchell Jones, Maximilian Haas-Heger · 3 authors totalMining Adverse Drug Reactions from Unstructured Mediums at Scale
Studies in computational intelligence · DOI 10.1007/978-3-031-14771-5_26 · 14 citations · Source: openalexDavid Talby, Hasham Ul Haq, Veysel Kocaman · 3 authors totalFederated Learning for Collaborative Financial Crimes Detection
Federated Learning · DOI 10.1007/978-3-030-96896-0_20 · 17 citations · Source: semantic-scholarNathalie Baracaldo, T. Suzumura, Yi Zhou, Ryo Kawahara, Heiko Ludwig · 5 authors totalLongitudinal changes of automated speech markers in MCI and mild AD
Alzheimer s & Dementia · DOI 10.1002/alz.061835 · 1 citations · Source: openalex+first-party-career-authorityMark Liberman, Sunghye Cho, Sanjana Shellikeri, Sharon Ash, Galit Agmon, Katheryn A Q Cousins, Carmen Gonzalez‐Recober, Naomi Nevler · 9 authors totalDemonstrate-Search-Predict: Composing retrieval and language models for knowledge-intensive NLP
arXiv.org · DOI 10.48550/arXiv.2212.14024 · arXiv 2212.14024 · 399 citations · Source: semantic-scholarRetrieval-augmented in-context learning has emerged as a powerful approach for addressing knowledge-intensive tasks using frozen language models (LM) and retrieval models (RM). Existing work has combined these in simple"retrieve-then-read"pipelines in which the RM retrieves passages that are inserted into the LM prompt. To begin to fully realize the potential of frozen LMs and RMs, we propose Demonstrate-Search-Predict (DSP), a framework that relies on passing natural language texts in sophisticated pipelines between an LM and an RM. DSP can express high-level programs that bootstrap pipeline-aware demonstrations, search for relevant passages, and generate grounded predictions, systematically breaking down problems into small transformations that the LM and RM can handle more reliably. We have written novel DSP programs for answering questions in open-domain, multi-hop, and conversational settings, establishing in early evaluations new state-of-the-art in-context learning results and delivering 37-120%, 8-39%, and 80-290% relative gains against the vanilla LM (GPT-3.5), a standard retrieve-then-read pipeline, and a contemporaneous self-ask pipeline, respectively. We release DSP at https://github.com/stanfordnlp/dsp
David Hall, Matei Zaharia, O. Khattab, Keshav Santhanam, Xiang Lisa Li, David Leo Wright Hall, Percy Liang, Christopher Potts · 8 authors totalLanguage models are better than humans at next-token prediction
Transactions on Machine Learning Research (2024) · DOI 10.48550/arXiv.2212.11281 · arXiv 2212.11281 · 21 citations · Source: arxiv+dblpCurrent language models are considered to have sub-human capabilities at natural language tasks like question-answering or writing code. However, language models are not trained to perform well at these tasks, they are trained to accurately predict the next token given previous tokes in tokenized text. It is not clear whether language models are better or worse than humans at next token prediction. To try to answer this question, we performed two distinct experiments to directly compare humans and language models on this front: one measuring top-1 accuracy and the other measuring perplexity. In both experiments, we find humans to be consistently \emph{worse} than even relatively small language models like GPT3-Ada at next-token prediction.
Buck Shlegeris, Fabien Roger, Lawrence Chan, Euan McLean · 4 authors totalLayoutDETR: Detection Transformer Is a Good Multimodal Layout Designer
European Conference on Computer Vision · DOI 10.48550/arXiv.2212.09877 · arXiv 2212.09877 · 21 citations · Source: arxiv+semantic-scholarGraphic layout designs play an essential role in visual communication. Yet handcrafting layout designs is skill-demanding, time-consuming, and non-scalable to batch production. Generative models emerge to make design automation scalable but it remains non-trivial to produce designs that comply with designers' multimodal desires, i.e., constrained by background images and driven by foreground content. We propose LayoutDETR that inherits the high quality and realism from generative modeling, while reformulating content-aware requirements as a detection problem: we learn to detect in a background image the reasonable locations, scales, and spatial relations for multimodal foreground elements in a layout. Our solution sets a new state-of-the-art performance for layout generation on public benchmarks and on our newly-curated ad banner dataset. We integrate our solution into a graphical system that facilitates user studies, and show that users prefer our designs over baselines by significant margins. Code, models, dataset, and demos are available at https://github.com/salesforce/LayoutDETR.
Ran Xu, Ning Yu, Chia-Chih Chen, Zeyuan Chen, Rui Meng, Ganglu Wu, P. Josel, Juan Carlos Niebles · 9 authors totalDiscovering Language Model Behaviors with Model-Written Evaluations
Annual Meeting of the Association for Computational Linguistics · DOI 10.48550/arXiv.2212.09251 · arXiv 2212.09251 · 991 citations · Source: semantic-scholarAs language models (LMs) scale, they develop many novel behaviors, good and bad, exacerbating the need to evaluate how they behave. Prior work creates evaluations with crowdwork (which is time-consuming and expensive) or existing data sources (which are not always available). Here, we automatically generate evaluations with LMs. We explore approaches with varying amounts of human effort, from instructing LMs to write yes/no questions to making complex Winogender schemas with multiple stages of LM-based generation and filtering. Crowdworkers rate the examples as highly relevant and agree with 90-100% of labels, sometimes more so than corresponding human-written datasets. We generate 154 datasets and discover new cases of inverse scaling where LMs get worse with size. Larger LMs repeat back a dialog user's preferred answer ("sycophancy") and express greater desire to pursue concerning goals like resource acquisition and goal preservation. We also find some of the first examples of inverse scaling in RL from Human Feedback (RLHF), where more RLHF makes LMs worse. For example, RLHF makes LMs express stronger political views (on gun rights and immigration) and a greater desire to avoid shut down. Overall, LM-written evaluations are high-quality and let us quickly discover many novel LM behaviors.
Tom Brown, Ethan Perez, Sam Ringer, Kamilė Lukošiūtė, Karina Nguyen, Edwin Chen, Scott Heiner, Craig Pettit · 63 authors totalConstitutional AI: Harmlessness from AI Feedback
arXiv.org · DOI 10.48550/arXiv.2212.08073 · arXiv 2212.08073 · 3,573 citations · Source: semantic-scholarAs AI systems become more capable, we would like to enlist their help to supervise other AIs. We experiment with methods for training a harmless AI assistant through self-improvement, without any human labels identifying harmful outputs. The only human oversight is provided through a list of rules or principles, and so we refer to the method as 'Constitutional AI'. The process involves both a supervised learning and a reinforcement learning phase. In the supervised phase we sample from an initial model, then generate self-critiques and revisions, and then finetune the original model on revised responses. In the RL phase, we sample from the finetuned model, use a model to evaluate which of the two samples is better, and then train a preference model from this dataset of AI preferences. We then train with RL using the preference model as the reward signal, i.e. we use 'RL from AI Feedback' (RLAIF). As a result we are able to train a harmless but non-evasive AI assistant that engages with harmful queries by explaining its objections to them. Both the SL and RL methods can leverage chain-of-thought style reasoning to improve the human-judged performance and transparency of AI decision making. These methods make it possible to control AI behavior more precisely and with far fewer human labels.
Tom Brown, Yuntao Bai, Saurav Kadavath, Sandipan Kundu, Amanda Askell, John Kernion, Andy Jones, A. Chen · 51 authors totalSurvey on social reputation mechanisms: Someone told me I can trust you
arXiv (Cornell University) · DOI 10.48550/arxiv.2212.06436 · 0 citations · Source: openalex+orcid+dblp-identityJohan Pouwelse, Thomas Werthenbach · 2 authors totalSuccessive Prompting for Decomposing Complex Questions
Conference on Empirical Methods in Natural Language Processing · DOI 10.48550/arXiv.2212.04092 · arXiv 2212.04092 · 154 citations · Source: semantic-scholarAnswering complex questions that require making latent decisions is a challenging task, especially when limited supervision is available. Recent works leverage the capabilities of large language models (LMs) to perform complex question answering in a few-shot setting by demonstrating how to output intermediate rationalizations while solving the complex question in a single pass. We introduce “Successive Prompting” where, we iteratively break down a complex task into a simple task, solve it, and then repeat the process until we get the final solution. Successive prompting decouples the supervision for decomposing complex questions from the supervision for answering simple questions, allowing us to (1) have multiple opportunities to query in-context examples at each reasoning step (2) learn question decomposition separately from question answering, including using synthetic data, and (3) use bespoke (fine-tuned) components for reasoning steps where a large LM does not perform well. The intermediate supervision is typically manually written, which can be expensive to collect. We introduce a way to generate synthetic dataset which can be used to bootstrap model’s ability to decompose and answer intermediate questions. Our best model (with successive prompting) achieves an improvement in F1 of ~5% when compared with a state-of-the-art model with synthetic augmentations and few-shot version of the DROP dataset.
Sameer Singh, Dheeru Dua, Shivanshu Gupta, Matt Gardner · 4 authors totalTackling Data Heterogeneity in Federated Learning with Class Prototypes
AAAI Conference on Artificial Intelligence · DOI 10.48550/arXiv.2212.02758 · arXiv 2212.02758 · 173 citations · Source: arxiv+semantic-scholarData heterogeneity across clients in federated learning (FL) settings is a widely acknowledged challenge. In response, personalized federated learning (PFL) emerged as a framework to curate local models for clients' tasks. In PFL, a common strategy is to develop local and global models jointly - the global model (for generalization) informs the local models, and the local models (for personalization) are aggregated to update the global model. A key observation is that if we can improve the generalization ability of local models, then we can improve the generalization of global models, which in turn builds better personalized models. In this work, we consider class imbalance, an overlooked type of data heterogeneity, in the classification setting. We propose FedNH, a novel method that improves the local models' performance for both personalization and generalization by combining the uniformity and semantics of class prototypes. FedNH initially distributes class prototypes uniformly in the latent space and smoothly infuses the class semantics into class prototypes. We show that imposing uniformity helps to combat prototype collapse while infusing class semantics improves local models. Extensive experiments were conducted on popular classification datasets under the cross-device setting. Our results demonstrate the effectiveness and stability of our method over recent works.
Ran Xu, Yutong Dai, Zeyuan Chen, Junnan Li, Shelby Heinecke, Lichao Sun · 6 authors totalMegaBlocks: Efficient Sparse Training with Mixture-of-Experts
Conference on Machine Learning and Systems · DOI 10.48550/arXiv.2211.15841 · arXiv 2211.15841 · 250 citations · Source: semantic-scholarWe present MegaBlocks, a system for efficient Mixture-of-Experts (MoE) training on GPUs. Our system is motivated by the limitations of current frameworks, which restrict the dynamic routing in MoE layers to satisfy the constraints of existing software and hardware. These formulations force a tradeoff between model quality and hardware efficiency, as users must choose between dropping tokens from the computation or wasting computation and memory on padding. To address these limitations, we reformulate MoE computation in terms of block-sparse operations and develop new block-sparse GPU kernels that efficiently handle the dynamism present in MoEs. Our approach never drops tokens and maps efficiently to modern hardware, enabling end-to-end training speedups of up to 40% over MoEs trained with the state-of-the-art Tutel library and 2.4x over DNNs trained with the highly-optimized Megatron-LM framework.
Matei Zaharia, Trevor Gale, D. Narayanan, C. Young, M. Zaharia · 5 authors totalWhat learning algorithm is in-context learning? Investigations with linear models.
CoRR · DOI 10.48550/arXiv.2211.15661 · arXiv 2211.15661 · Source: dblp+stanford-authorityTengyu Ma, Ekin Akyürek, Dale Schuurmans, Jacob Andreas, Tengyu Ma 0001, Denny Zhou · 6 authors totalA Theoretical Study of Inductive Biases in Contrastive Learning.
CoRR · DOI 10.48550/arXiv.2211.14699 · arXiv 2211.14699 · Source: dblp+stanford-authorityTengyu Ma, Jeff Z. HaoChen, Tengyu Ma 0001 · 3 authors totalFirst Steps Toward Understanding the Extrapolation of Nonlinear Models to Unseen Domains.
CoRR · DOI 10.48550/arXiv.2211.11719 · arXiv 2211.11719 · Source: dblp+stanford-authorityTengyu Ma, Kefan Dong, Tengyu Ma 0001 · 3 authors totalHow Does Sharpness-Aware Minimization Minimize Sharpness?
CoRR · DOI 10.48550/arXiv.2211.05729 · arXiv 2211.05729 · Source: dblp+stanford-authorityTengyu Ma, Kaiyue Wen, Tengyu Ma 0001, Zhiyuan Li 0005 · 4 authors totalOn Optimizing the Communication of Model Parallelism
MLSys 2023 · DOI 10.48550/arXiv.2211.05322 · arXiv 2211.05322 · 53 citations · Source: arxiv+semantic-scholarWe study a novel and important communication pattern in large-scale model-parallel deep learning (DL), which we call cross-mesh resharding. This pattern emerges when the two paradigms of model parallelism - intra-operator and inter-operator parallelism - are combined to support large models on large clusters. In cross-mesh resharding, a sharded tensor needs to be sent from a source device mesh to a destination device mesh, on which the tensor may be distributed with the same or different layouts. We formalize this as a many-to-many multicast communication problem, and show that existing approaches either are sub-optimal or do not generalize to different network topologies or tensor layouts, which result from different model architectures and parallelism strategies. We then propose two contributions to address cross-mesh resharding: an efficient broadcast-based communication system, and an"overlapping-friendly"pipeline schedule. On microbenchmarks, our overall system outperforms existing ones by up to 10x across various tensor and mesh layouts. On end-to-end training of two large models, GPT-3 and U-Transformer, we improve throughput by 10% and 50%, respectively.
Zhuohan Li, Yonghao Zhuang, Hexu Zhao, Lianmin Zheng, Eric P. Xing, Qirong Ho, Joseph E. Gonzalez, Ion Stoica · 9 authors totalRecD: Deduplication for End-to-End Deep Learning Recommendation Model Training Infrastructure
arXiv · DOI 10.48550/arxiv.2211.05239 · arXiv 2211.05239 · 4 citations · Source: openalexWe present RecD (Recommendation Deduplication), a suite of end-to-end infrastructure optimizations across the Deep Learning Recommendation Model (DLRM) training pipeline. RecD addresses immense storage, preprocessing, and training overheads caused by feature duplication inherent in industry-scale DLRM training datasets. Feature duplication arises because DLRM datasets are generated from interactions. While each user session can generate multiple training samples, many features' values do not change across these samples. We demonstrate how RecD exploits this property, end-to-end, across a deployed training pipeline. RecD optimizes data generation pipelines to decrease dataset storage and preprocessing resource demands and to maximize duplication within a training batch. RecD introduces a new tensor format, InverseKeyedJaggedTensors (IKJTs), to deduplicate feature values in each batch. We show how DLRM model architectures can leverage IKJTs to drastically increase training throughput. RecD improves the training and preprocessing throughput and storage efficiency by up to 2.48x, 1.79x, and 3.71x, respectively, in an industry-scale DLRM training system.
Dhruv Choudhary, Mark Zhao, Devashish Tyagi, Ajay Somani, Max Kaplan, Sung‐Han Lin, Pumma, Sarunya, Jongsoo Park · 12 authors totalMeasuring Progress on Scalable Oversight for Large Language Models
arXiv.org · DOI 10.48550/arXiv.2211.03540 · arXiv 2211.03540 · 235 citations · Source: semantic-scholarDeveloping safe and useful general-purpose AI systems will require us to make progress on scalable oversight: the problem of supervising systems that potentially outperform us on most skills relevant to the task at hand. Empirical work on this problem is not straightforward, since we do not yet have systems that broadly exceed our abilities. This paper discusses one of the major ways we think about this problem, with a focus on ways it can be studied empirically. We first present an experimental design centered on tasks for which human specialists succeed but unaided humans and current general AI systems fail. We then present a proof-of-concept experiment meant to demonstrate a key feature of this experimental design and show its viability with two question-answering tasks: MMLU and time-limited QuALITY. On these tasks, we find that human participants who interact with an unreliable large-language-model dialog assistant through chat -- a trivial baseline strategy for scalable oversight -- substantially outperform both the model alone and their own unaided performance. These results are an encouraging sign that scalable oversight will be tractable to study with present models and bolster recent findings that large language models can productively assist humans with difficult tasks.
Tom Brown, Sam Bowman, Jeeyoon Hyun, Ethan Perez, Edwin Chen, Craig Pettit, Scott Heiner, Kamilė Lukošiūtė · 46 authors totalInterpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 small
ICLR 2023 · DOI 10.48550/arXiv.2211.00593 · arXiv 2211.00593 · 1,205 citations · Source: arxiv+dblpResearch in mechanistic interpretability seeks to explain behaviors of machine learning models in terms of their internal components. However, most previous work either focuses on simple behaviors in small models, or describes complicated behaviors in larger models with broad strokes. In this work, we bridge this gap by presenting an explanation for how GPT-2 small performs a natural language task called indirect object identification (IOI). Our explanation encompasses 26 attention heads grouped into 7 main classes, which we discovered using a combination of interpretability approaches relying on causal interventions. To our knowledge, this investigation is the largest end-to-end attempt at reverse-engineering a natural behavior "in the wild" in a language model. We evaluate the reliability of our explanation using three quantitative criteria--faithfulness, completeness and minimality. Though these criteria support our explanation, they also point to remaining gaps in our understanding. Our work provides evidence that a mechanistic understanding of large ML models is feasible, opening opportunities to scale our understanding to both larger models and more complex tasks.
Buck Shlegeris, Kevin Wang, Alexandre Variengien, Arthur Conmy, Jacob Steinhardt · 5 authors totalSame Pre-training Loss, Better Downstream: Implicit Bias Matters for Language Models.
CoRR · DOI 10.48550/arXiv.2210.14199 · arXiv 2210.14199 · Source: dblp+stanford-authorityTengyu Ma, Hong Liu, Sang Michael Xie, Zhiyuan Li 0005, Tengyu Ma 0001 · 5 authors totalNVIDIA FLARE: Federated Learning from Simulation to Real-World
IEEE Data Engineering Bulletin · DOI 10.48550/arXiv.2210.13291 · arXiv 2210.13291 · 181 citations · Source: arxiv+semantic-scholarFederated learning (FL) enables building robust and generalizable AI models by leveraging diverse datasets from multiple collaborators without centralizing the data. We created NVIDIA FLARE as an open-source software development kit (SDK) to make it easier for data scientists to use FL in their research and real-world applications. The SDK includes solutions for state-of-the-art FL algorithms and federated machine learning approaches, which facilitate building workflows for distributed learning across enterprises and enable platform developers to create a secure, privacy-preserving offering for multiparty collaboration utilizing homomorphic encryption or differential privacy. The SDK is a lightweight, flexible, and scalable Python package. It allows researchers to apply their data science workflows in any training libraries (PyTorch, TensorFlow, XGBoost, or even NumPy) in real-world FL settings. This paper introduces the key design principles of NVFlare and illustrates some use cases (e.g., COVID analysis) with customizable FL workflows that implement different privacy-preserving algorithms. Code is available at https://github.com/NVIDIA/NVFlare.
Chester Chen, Holger R. Roth, Yan Cheng, Yuhong Wen, Isaac Yang, Ziyue Xu, Yuan-Ting Hsieh, Kristopher Kersten · 23 authors totalAn Empirical Study on Finding Spans
Conference on Empirical Methods in Natural Language Processing · DOI 10.48550/arXiv.2210.06824 · arXiv 2210.06824 · 5 citations · Source: semantic-scholarWe present an empirical study on methods for span finding, the selection of consecutive tokens in text for some downstream tasks. We focus on approaches that can be employed in training end-to-end information extraction systems, and find there is no definitive solution without considering task properties, and provide our observations to help with future design choices: 1) a tagging approach often yields higher precision while span enumeration and boundary prediction provide higher recall; 2) span type information can benefit a boundary prediction approach; 3) additional contextualization does not help span finding in most cases.
Tongfei Chen, Weiwei Gu, Boyuan Zheng, Yunmo Chen, Benjamin Van Durme · 5 authors totalIterative Document-level Information Extraction via Imitation Learning
Conference of the European Chapter of the Association for Computational Linguistics · DOI 10.48550/arXiv.2210.06600 · arXiv 2210.06600 · 15 citations · Source: semantic-scholarWe present a novel iterative extraction model, IterX, for extracting complex relations, or templates, i.e., N-tuples representing a mapping from named slots to spans of text within a document. Documents may feature zero or more instances of a template of any given type, and the task of template extraction entails identifying the templates in a document and extracting each template’s slot values. Our imitation learning approach casts the problem as a Markov decision process (MDP), and relieves the need to use predefined template orders to train an extractor. It leads to state-of-the-art results on two established benchmarks – 4-ary relation extraction on SciREX and template extraction on MUC-4 – as well as a strong baseline on the new BETTER Granular task.
Tongfei Chen, Yunmo Chen, William Gantt Walden, Weiwei Gu, Aaron Steven White, Benjamin Van Durme · 6 authors totalContrastive Training Improves Zero-Shot Classification of Semi-structured Documents
Annual Meeting of the Association for Computational Linguistics · DOI 10.48550/arXiv.2210.05613 · arXiv 2210.05613 · 5 citations · Source: semantic-scholarWe investigate semi-structured document classification in a zero-shot setting. Classification of semi-structured documents is more challenging than that of standard unstructured documents, as positional, layout, and style information play a vital role in interpreting such documents. The standard classification setting where categories are fixed during both training and testing falls short in dynamic environments where new document categories could potentially emerge. We focus exclusively on the zero-shot setting where inference is done on new unseen classes. To address this task, we propose a matching-based approach that relies on a pairwise contrastive objective for both pretraining and fine-tuning. Our results show a significant boost in Macro F$_1$ from the proposed pretraining step in both supervised and unsupervised zero-shot settings.
Sunil Mallya, Muhammad Khalifa, Yogarshi Vyas, Shuai Wang, Graham Horwood, S. Mallya, Miguel Ballesteros · 7 authors totalTrustVault: A privacy-first data wallet for the European Blockchain Services Infrastructure
arXiv (Cornell University) · DOI 10.48550/arxiv.2210.02987 · 0 citations · Source: openalex+orcid+dblp-identityJohan Pouwelse, Sharif Jacobino · 2 authors totalPolysemanticity and Capacity in Neural Networks
arXiv preprint · DOI 10.48550/arXiv.2210.01892 · arXiv 2210.01892 · 74 citations · Source: arxiv+dblpIndividual neurons in neural networks often represent a mixture of unrelated features. This phenomenon, called polysemanticity, can make interpreting neural networks more difficult and so we aim to understand its causes. We propose doing so through the lens of feature \emph{capacity}, which is the fractional dimension each feature consumes in the embedding space. We show that in a toy model the optimal capacity allocation tends to monosemantically represent the most important features, polysemantically represent less important features (in proportion to their impact on the loss), and entirely ignore the least important features. Polysemanticity is more prevalent when the inputs have higher kurtosis or sparsity and more prevalent in some architectures than others. Given an optimal allocation of capacity, we go on to study the geometry of the embedding space. We find a block-semi-orthogonal structure, with differing block sizes in different models, highlighting the impact of model architecture on the interpretability of its neurons.
Buck Shlegeris, Adam Scherlis, Kshitij Sachan, Adam S. Jermyn, Joe Benton · 5 authors totalMake-A-Video: Text-to-Video Generation without Text-Video Data
ICLR 2023 · arXiv 2209.14792 · 2,149 citations · Source: semantic-scholarWe propose Make-A-Video -- an approach for directly translating the tremendous recent progress in Text-to-Image (T2I) generation to Text-to-Video (T2V). Our intuition is simple: learn what the world looks like and how it is described from paired text-image data, and learn how the world moves from unsupervised video footage. Make-A-Video has three advantages: (1) it accelerates training of the T2V model (it does not need to learn visual and multimodal representations from scratch), (2) it does not require paired text-video data, and (3) the generated videos inherit the vastness (diversity in aesthetic, fantastical depictions, etc.) of today's image generation models. We design a simple yet effective way to build on T2I models with novel and effective spatial-temporal modules. First, we decompose the full temporal U-Net and attention tensors and approximate them in space and time. Second, we design a spatial temporal pipeline to generate high resolution and frame rate videos with a video decoder, interpolation model and two super resolution models that can enable various applications besides T2V. In all aspects, spatial and temporal resolution, faithfulness to text, and quality, Make-A-Video sets the new state-of-the-art in text-to-video generation, as determined by both qualitative and quantitative measures.
Sonal Gupta, Uriel Singer, Adam Polyak, Thomas Hayes, Xiaoyue Yin, Jie An, Songyang Zhang, Qiyuan Hu · 13 authors totalIn-context Learning and Induction Heads
arXiv.org · DOI 10.48550/arXiv.2209.11895 · arXiv 2209.11895 · 986 citations · Source: semantic-scholar"Induction heads"are attention heads that implement a simple algorithm to complete token sequences like [A][B] ... [A] ->[B]. In this work, we present preliminary and indirect evidence for a hypothesis that induction heads might constitute the mechanism for the majority of all"in-context learning"in large transformer models (i.e. decreasing loss at increasing token indices). We find that induction heads develop at precisely the same point as a sudden sharp increase in in-context learning ability, visible as a bump in the training loss. We present six complementary lines of evidence, arguing that induction heads may be the mechanistic source of general in-context learning in transformer models of any size. For small attention-only models, we present strong, causal evidence; for larger models with MLPs, we present correlational evidence.
Tom Brown, Catherine Olsson, Nelson Elhage, Neel Nanda, Nicholas Joseph, Nova Dassarma, T. Henighan, Benjamin Mann · 26 authors totalAssessing ASR Model Quality on Disordered Speech using BERTScore.
CoRR · DOI 10.48550/arxiv.2209.10591 · arXiv 2209.10591 · Source: dblpKatrin Tomanek, Jimmy Tobin, Qisheng Li, Subhashini Venugopalan, Katie Seaver, Richard Cave · 6 authors totalRed Teaming Language Models to Reduce Harms: Methods, Scaling Behaviors, and Lessons Learned
arXiv.org · DOI 10.48550/arXiv.2209.07858 · arXiv 2209.07858 · 868 citations · Source: semantic-scholarWe describe our early efforts to red team language models in order to simultaneously discover, measure, and attempt to reduce their potentially harmful outputs. We make three main contributions. First, we investigate scaling behaviors for red teaming across 3 model sizes (2.7B, 13B, and 52B parameters) and 4 model types: a plain language model (LM); an LM prompted to be helpful, honest, and harmless; an LM with rejection sampling; and a model trained to be helpful and harmless using reinforcement learning from human feedback (RLHF). We find that the RLHF models are increasingly difficult to red team as they scale, and we find a flat trend with scale for the other model types. Second, we release our dataset of 38,961 red team attacks for others to analyze and learn from. We provide our own analysis of the data and find a variety of harmful outputs, which range from offensive language to more subtly harmful non-violent unethical outputs. Third, we exhaustively describe our instructions, processes, statistical methodologies, and uncertainty about red teaming. We hope that this transparency accelerates our ability to work together as a community in order to develop shared norms, practices, and technical standards for how to red team language models.
Tom Brown, Deep Ganguli, Liane Lovitt, John Kernion, Amanda Askell, Yuntao Bai, Saurav Kadavath, Benjamin Mann · 36 authors totalFuture Gradient Descent for Adapting the Temporal Shifting Data Distribution in Online Recommendation Systems
arXiv · DOI 10.48550/arxiv.2209.01143 · arXiv 2209.01143 · 0 citations · Source: openalexOne of the key challenges of learning an online recommendation model is the temporal domain shift, which causes the mismatch between the training and testing data distribution and hence domain generalization error. To overcome, we propose to learn a meta future gradient generator that forecasts the gradient information of the future data distribution for training so that the recommendation model can be trained as if we were able to look ahead at the future of its deployment. Compared with Batch Update, a widely used paradigm, our theory suggests that the proposed algorithm achieves smaller temporal domain generalization error measured by a gradient variation term in a local regret. We demonstrate the empirical advantage by comparing with various representative baselines.
Dhruv Choudhary, Mao Ye, Ruichen Jiang, Haoxiang Wang, Xiaocong Du, Bhargav Bhushanam, Aryan Mokhtari, Arun Kejariwal · 9 authors totalApiary: A DBMS-Integrated Transactional Function-as-a-Service Framework
arXiv (Cornell University) · DOI 10.48550/arxiv.2208.13068 · 4 citations · Source: openalex+authoritative-profilePeter Bailis, Peter Kraft, Qian Li, Kostis Kaffes, Athinagoras Skiadopoulos, Deeptaanshu Kumar, Danny I. Cho, Jason Li · 19 authors totalTAG: Boosting Text-VQA via Text-aware Visual Question-answer Generation
British Machine Vision Conference · DOI 10.48550/arXiv.2208.01813 · arXiv 2208.01813 · 26 citations · Source: arxiv+semantic-scholarText-VQA aims at answering questions that require understanding the textual cues in an image. Despite the great progress of existing Text-VQA methods, their performance suffers from insufficient human-labeled question-answer (QA) pairs. However, we observe that, in general, the scene text is not fully exploited in the existing datasets -- only a small portion of the text in each image participates in the annotated QA activities. This results in a huge waste of useful information. To address this deficiency, we develop a new method to generate high-quality and diverse QA pairs by explicitly utilizing the existing rich text available in the scene context of each image. Specifically, we propose, TAG, a text-aware visual question-answer generation architecture that learns to produce meaningful, and accurate QA samples using a multimodal transformer. The architecture exploits underexplored scene text information and enhances scene understanding of Text-VQA models by combining the generated QA pairs with the initial training data. Extensive experimental results on two well-known Text-VQA benchmarks (TextVQA and ST-VQA) demonstrate that our proposed TAG effectively enlarges the training data that helps improve the Text-VQA performance without extra labeling effort. Moreover, our model outperforms state-of-the-art approaches that are pre-trained with extra large-scale data. Code is available at https://github.com/HenryJunW/TAG.
Ran Xu, Jun Wang, Mingfei Gao, Yuqian Hu, Ramprasaath R. Selvaraju, Chetan Ramaiah, J. JáJá, Larry S. Davis · 8 authors totalEfficient Graph-Friendly COCO Metric Computation for Train-Time Model Evaluation
arXiv · DOI 10.48550/arxiv.2207.12120 · arXiv 2207.12120 · 12 citations · Source: semantic-scholarEvaluating the COCO mean average precision (MaP) and COCO recall metrics as part of the static computation graph of modern deep learning frameworks poses a unique set of challenges. These challenges include the need for maintaining a dynamic-sized state to compute mean average precision, reliance on global dataset-level statistics to compute the metrics, and managing differing numbers of bounding boxes between images in a batch. As a consequence, it is common practice for researchers and practitioners to evaluate COCO metrics as a post training evaluation step. With a graph-friendly algorithm to compute COCO Mean Average Precision and recall, these metrics could be evaluated at training time, improving visibility into the evolution of the metrics through training curve plots, and decreasing iteration time when prototyping new model versions. Our contributions include an accurate approximation algorithm for Mean Average Precision, an open source implementation of both COCO mean average precision and COCO recall, extensive numerical benchmarks to verify the accuracy of our implementations, and an open-source training loop that include train-time evaluation of mean average precision and recall.
Francois Chollet, Luke Wood, François Chollet · 3 authors totalCalibrated ensembles can mitigate accuracy tradeoffs under distribution shift.
CoRR · DOI 10.48550/arXiv.2207.08977 · arXiv 2207.08977 · Source: dblp+stanford-authorityTengyu Ma, Ananya Kumar, Tengyu Ma 0001, Percy Liang, Aditi Raghunathan · 5 authors totalFederated Unlearning: How to Efficiently Erase a Client in FL?
arXiv.org · arXiv 2207.05521 · 221 citations · Source: semantic-scholar+arxivWith privacy legislation empowering the users with the right to be forgotten, it has become essential to make a model amenable for forgetting some of its training data. However, existing unlearning methods in the machine learning context can not be directly applied in the context of distributed settings like federated learning due to the differences in learning protocol and the presence of multiple actors. In this paper, we tackle the problem of federated unlearning for the case of erasing a client by removing the influence of their entire local data from the trained global model. To erase a client, we propose to first perform local unlearning at the client to be erased, and then use the locally unlearned model as the initialization to run very few rounds of federated learning between the server and the remaining clients to obtain the unlearned global model. We empirically evaluate our unlearning method by employing multiple performance measures on three datasets, and demonstrate that our unlearning method achieves comparable performance as the gold standard unlearning method of federated retraining from scratch, while being significantly efficient. Unlike prior works, our unlearning method neither requires global access to the data used for training nor the history of the parameter updates to be stored by the server or any of the clients.
Nathalie Baracaldo, Anisa Halimi, S. Kadhe, Ambrish Rawat · 4 authors totalLanguage Models (Mostly) Know What They Know
arXiv.org · DOI 10.48550/arXiv.2207.05221 · arXiv 2207.05221 · 1,908 citations · Source: semantic-scholarWe study whether language models can evaluate the validity of their own claims and predict which questions they will be able to answer correctly. We first show that larger models are well-calibrated on diverse multiple choice and true/false questions when they are provided in the right format. Thus we can approach self-evaluation on open-ended sampling tasks by asking models to first propose answers, and then to evaluate the probability"P(True)"that their answers are correct. We find encouraging performance, calibration, and scaling for P(True) on a diverse array of tasks. Performance at self-evaluation further improves when we allow models to consider many of their own samples before predicting the validity of one specific possibility. Next, we investigate whether models can be trained to predict"P(IK)", the probability that"I know"the answer to a question, without reference to any particular proposed answer. Models perform well at predicting P(IK) and partially generalize across tasks, though they struggle with calibration of P(IK) on new tasks. The predicted P(IK) probabilities also increase appropriately in the presence of relevant source materials in the context, and in the presence of hints towards the solution of mathematical word problems. We hope these observations lay the groundwork for training more honest models, and for investigating how honesty generalizes to cases where models are trained on objectives other than the imitation of human writing.
Tom Brown, Saurav Kadavath, Tom Conerly, Amanda Askell, T. Henighan, Dawn Drain, Ethan Perez, Nicholas Schiefer · 36 authors totalGT4SD: Generative Toolkit for Scientific Discovery
arXiv.org · DOI 10.48550/arXiv.2207.03928 · 6 citations · Source: semantic-scholarAlexy Khrabrov, Matteo Manica, Joris Cadow, Dimitrios Christofidellis, A. Dave, Jannis Born, D. Clarke, Yves Gaëtan Nana Teukam · 24 authors totalManu: A Cloud Native Vector Database Management System
arXiv · arXiv 2206.13843 · Source: arxiv+zillizPresents Manu, a cloud-native vector database designed for evolvability, tunable consistency, elasticity, and high performance.
Frank Liu, Rentong Guo, Xiaofan Luan, Long Xiang, Xiao Yan, Xiaomeng Yi, Jigao Luo, Qianya Cheng · 15 authors total