当前位置：首页 > 编程日记 > 正文

如何在3天内拿下DigSci大赛的亚军？| DigSci科学数据挖掘大赛

编程日记 2024-12-19 02:20:00

作者 | 朱翔宇(DOTA)

来源 | AI算法之心（AIHeartForYou）

【导读】本文将基于作者在中国计算机大会方案宣讲PPT中的内容，给出此次比赛的完整方案，主要从赛题理解、建模核心思路、算法核心思想等角度阐述，同时对相关细节进行了适当的拓展讲解。涉及SIF Sentence Embedding、InferSent、Bi-LSTM和BERT等算法及相关概念。

任务背景

640?wx_fmt=png

科学研究已经成为现代社会创新的主要动力。大量科研数据的积累也让我们可以理解和预测科研发展，并能用来指导未来的研究。论文是人类最前沿知识的媒介，因此如果可以理解论文中的数据，可以极大地扩充计算机理解知识的能力和范围。

在论文中，作者经常会引用其他论文，并对被引论文做出对应描述。如果我们可以自动地理解、识别描述对应的被引论文，不仅可以加深对科研脉络的理解，还能在科研知识图谱、科研自动问答系统和自动摘要系统等领域有所进步。

任务描述

本次比赛将提供一个论文库（约含20万篇论文），同时提供对论文的描述段落，来自论文中对同类研究的介绍。参赛选手需要为描述段落匹配三篇最相关的论文。

例子：

描述段落：

An efficient implementation based on BERT [1] and graph neural network (GNN) [2] is introduced.

相关论文：

[1] BERT: Pre-training of deep bidirectional transformers for language understanding.

[2] Relational inductive biases, deep learning, and graph networks.

评测方案

准确率（Precision）: 提交结果的准确性通过 Mean Average Precision @ 3 (MAP@3) 打分，具体公式如下：

640?wx_fmt=png

其中，|U|是需要预测的press_id总个数，P(k)是在k处的精度，n是paper个数。

题意分析

从任务描述中我们可以看到，该任务需要对描述段落匹配三篇最相关的论文。单从形式上可以理解为这是一个“完形填空”任务。但相较于在本文的相应位置上填上相应的词语不同的是，这里需要填充的是一个Sentence，也就是论文的题目。但是如果你按照这个思路去寻求解决方案，你会发现在这个量级的文本数据上，一般算力是满足不了的。

既然如此，那我们不如换一个思路来思考这个问题，“对描述段落匹配三篇最相关的论文”，其实最简单的实现方式是计算描述段落和论文库里所有论文的相似度，找出最相似的即可。但这同样会存在一个问题，通过对数据进行探查你会发现“An efficient implementation based on BERT [1] and graph neural network (GNN) [2] is introduced.”这一描述段落，同时引用了两篇文章，那么在计算相似度时，到底哪个位置该是哪篇文章呢？

基于分析中提到的问题和难点，本次比赛给出了以下解决方案。

建模核心思路

640?wx_fmt=png

在解决本问题时，我借鉴了推荐算法的思想，将问题拆解了两部分——召回和排序。在召回阶段，使用了两种方式，其一是利用Word2Vec和TFIDF方法，将描述段落利用Word2Vec得到每个词的词向量，同时对句子中的词使用TF-IDF为权重进行加权得到Sentence Embedding，同时为了得到更好的效果，这里做了一个改进，即使用Smooth Inverse Frequency代替TFIDF作为每个词的权重；其二是利用TFIDF得到Sentence Embedding。两种方法各自计算余弦相似度得到3篇论文，去重后召回集中每个段落有3-6篇不等的召回论文。

在排序阶段，我们利用Bi-LSTM对描述段落Description和论文文本PaperText组成句子对（Description，PaperText）进行编码，在中间层对两个向量进行差值和内积操作后，在输出层经过Dense和Softmax层后得到概率值后排序。

算法核心思想

1.SIF Sentence Embedding

640?wx_fmt=png

SIF Sentence Embedding 使用预训练好的词向量，使用加权平均的方法，对句子中所有词对应的词向量进行计算，得到整个句子的embedding向量。

SIF 的计算分为两步：

1）对句子中的每个词向量，乘以一个独特的权重b，权重b是一个常数 a除以a与该词频率的和，这种做法的会对出现频率高词进行降权，也就是说出现频次越高，其权重也就越小；

2）计算句向量矩阵的第一主成分u，让每个Sentence Embedding减去它在u上的投影，具体参见论文[2]；

这里，利用该方法做召回，在验证集上的准确性要比其他两种方式效果好。

（1）对句子中所有单词求平均得到sentence embedding；

（2）对句子中所有单词利用TF-IDF值加权后求平均得到sentence embedding。

2. InferSent

640?wx_fmt=png

InferSent相似度模型是Facebook提出了一种通过不同的encoder得到Sentence Embedding，然后计算两个Sentence Embedding的差值、点乘得到交互向量，计算两者之间的相似度。

这里，对原始论文方法做了两处修改：

其一，针对这个问题对3-way softmax层（entailment，contradiction，neutral）做了相应的修改变为2-way softmax；

其二，中间层去掉了u和v，只使用差值和内积两种特征表征方式；

640?wx_fmt=png

同时在7中编码器：1）LSTM, 2）GRU, 3）bi-GRU, 4）bi-LSTM(mean pooling), 5）bi-LSTM(max pooling), 6）self-attention, 7）CNN 中选用了Bi-LSTM MaxPooling的方式。

3. Others-BERT

640?wx_fmt=png

当时，在BERT时代，解决方案的尝试中少不了它的身影，这里我们用BERT尝试了两种方案，其一是利用BERT对Description和PaperText进行编码后，计算文本的余弦相似度；其二是在上述整体模型中，用BERT替换InferSent部分。

以上两种方式是我和何从庆在CCL“中国法研杯”相似案例匹配评测竞赛中使用的方案（https://zhuanlan.zhihu.com/p/88207736），但在本任务中首次尝试后发现线下Valid是不Work的，因为本次比赛我只做了最后两天，5次线上提交后得到终榜最高分，所以未对其他更多方式做尝试性修正，如果BERT在本任务上work的话，按照经验应该可以提升不少MAP@3，有兴趣的朋友可以尝试一下。

总结

入职两年，见过了许多形形色色，感慨良多，但初心没忘，为了让自己保持一个良好的竞技状态和学习状态。每次任务又是一个不断审视、总结自己的过程。无论是结构化数据、自然语言处理领域还是图像领域，欢迎志同道合的小伙伴一起分享探讨，共同进步。

参考论文：

[1] Sanjeev Arora, Yingyu Liang, Tengyu Ma. A Simple but Tough-to-Beat Baseline for Sentence Embeddings, ICLR 2017

[2] Conneau A , Kiela D , Schwenk H , et al. Supervised Learning of Universal Sentence Representations from Natural Language Inference Data[J]. 2017.

[3] Devlin J , Chang M W , Lee K , et al. BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding[J]. 2018.

（*本文为 AI科技大本营投稿文章，请联系作者）

◆

精彩推荐

◆

2019 中国大数据技术大会（BDTC）再度来袭！豪华主席阵容及百位技术专家齐聚，15 场精选专题技术和行业论坛，超强干货+技术剖析+行业实践立体解读，深入解析热门技术在行业中的实践落地。5 折票倒计时 7 天！

如何在3天内拿下DigSci大赛的亚军？| DigSci科学数据挖掘大赛

推荐阅读

相关文章：

如何获得Oracle系统性能统计？ Oracle大型数据库系统在AIXUNIX上的实战详解集中讨论 14...

百度搜索查询命令——组合型

Delphi多线程编程中的技巧

Ubuntu14.04下配置OpenGL及测试代码

VarGFaceNet：地平线提出轻量级、有效可变组卷积的人脸识别网络

【TortoiseSVN使用教程】

[ASP.NET MVC 小牛之路]10 - Controller 和 Action (2)

Linux下遍历指定目录的C++实现

要活102年，阿里凭借的是什么？

Go基础知识学习(6) 接口

多模态数据+知识图谱，这次你的疑难杂症有解了！

Linux下通过gettimeofday函数获取程序段执行时间

matlab练习程序（Log Polar变换）

SiteMapPath基本用法

closurescallbacks

吴恩达老师深度学习视频课笔记：卷积神经网络

COCO 2019挑战赛，旷视研究院拿下三项计算机识别冠军 | ICCV 2019

runners学习

Swift基础 - - 高德地图实践

Shell脚本示例代码

「2019嵌入式智能国际大会」 399元超值学生票来啦，帮你豪省2600元！

ThinkPHP学习笔记之Model操作

Ubuntu 14.04 64位上配置JDK操作步骤

Swift语言实现代理传值

亚马逊马超：如何使用DGL进行大规模图神经网络训练？

Python学习系列（六）(模块)

Ubuntu14.04 64位上配置终端显示git分支名称

创建UITextField对象

non-local神经网络：通过非局部操作解决深度神经网络核心问题

fgets()用法笔记