Gated Self-Matching Networks for Reading Comprehension and Question Answering论文笔记

本文提出了一种基于端到端多层神经网络的阅读理解式问答系统模型,该模型在Stanford SQuAD数据集上取得领先结果。模型通过编码问题和篇章,使用门注意力机制和自匹配注意力机制,最后利用Pointer-network预测答案边界。

原文下载链接

摘要

检索式问答系统试图从文档中获取问题的答案。一般步骤是先从一众文档中检索相关文档,然后再进一步检索文档回答问题。本文解决的是后一步,即阅读理解式的问答系统。文章基于端到端的多层神经网络模型从篇章中获取答案。

模型分为四部分:一是使用多层双向神经网络编码问题和篇章的语义向量表示;二是使用门注意力机制得到问题感知的篇章的语义向量表示;三是通过 Self-Matching 注意力机制提炼篇章的语义向量表示,从全部篇章中编码最终语义向量表示;四是利用 Pointer-network 来预测答案边界,从而得到最终答案。在 Stanford 发布的机器阅读理解比赛数据集 SQuAD 上,本文提出的模型的单模型和集成模型结果都分别排名第一。

Task 描述

对于阅读理解式的问答系统,给出了一个段落P和问题Q,我们的任务是根据在P中找到的信息预测答案A.

Gated Self-Matching Networks(门自匹配网络)

整个框架如Figure 1所示,首先,问题和段落分别由双向循环网络处理。 然后,我们将问题和段落通过门注意力的循环网络(gated attention-based recurrent networks)相匹配,从而获得该段落的问题感知表示。 之后,用自匹配注意力来整和整个段落的信息以进一步优化段落的表达,然后将其输入到输出层来预测答案的边界。

                 

1.问题与段落编码

问题,段落,首先将words转换成word-level的embeddings以及character-level的embeddings,character-level的embeddings是双向RNN的最后一个隐含状态,之所以加上character-level的embeddings,是因为其能够处理out-of-vocab(OOV)的情况,通过双向RNN(所有网络中的RNN都是采用的GRU,因为GRU和lstm的效果相当,但是计算量较小)得到Q和P的新的表达:                                        

2.门注意力RNN(Gated Attention-based Recurrent Networks)

这一层的目的是将问题中的信息添加到段落的表达中:

                                                         

                                                   

                                                              注括号内为[u_t_(P),c_t]*

V_t_(P)为新生产的段落中words的表达,请参看Figure1进行理解,门是基于当前段落word及其 attention-pooling vector of the question,其侧重于问题与当前段落word之间的关系。 门有效地模拟了阅读理解中只有部分word是和问题相关的。

3.自匹配注意力(self-Matching Attention)

通过上述操作,已经编码出了段落中的哪部分是重要的,但是这样的编码还存在的问题是:这部分重要段落缺乏上下文信息,因此这部分就是解决该问题的。该部分编码出的段落表达(请参看Figure1):

                                                             

                                                     

4.输出层

采用pointer network来预测答案的开始和结束位置,利用注意力机制作为选择起始位置()和结束位置():

                                                           

其中为pointer network的最后一个隐含层状态,pointer network的输入为基于当前预测概率的attention-pooling vector:

                                                            

采用问题向量作为attention-pooling vector:

                                              

训练时,最小化真实start和end位置的负log概率之和。

实验

文章的实验做得比较充分,这里补充一些实现细节:out-of-vocab的word是用零向量表示的;采用1层双向GRU来计算character-level embeddings,3层双向RGU编码问题和段落,所有层中隐含层向量长度都设置为75,层间dropout 率为0.2,优化方法AdaDelta,初始学习率为1,分别为0.95和1e-6。

1.和其他算法的对比实验

R-NET代表本文算法

2.每个部分的重要性

-代表去除,+代表添加,上述两个表格可以反映出本文各个部分的重要作用

3.统计发现

                                  

(a)可以看出问题类型为when和who时,表现较好,当为why时表现很差,这主要是因为why问题的答案是多样的;(b)可以看出,当答案很长时,表现很差,这说明答案越长越难预测;(c)和(d)可以看出整体相对稳定,当段落和问题很长时,可能表现较差,这是因为其中的有用信息所占比例较短。

 

<think>好的,我现在要帮助用户了解或实现Medical Transformer模型中的Gated Axial-Attention机制,特别是用于医学图像分割的部分。首先,我需要回忆一下用户提供的引用内容。用户引用了两篇资料,其中引用[1]是关于Medical Transformer的论文和代码库,而引用[2]列出了其他相关的分割模型,但用户主要关注的是Medical Transformer。 用户的问题具体是关于Gated Axial-Attention的实现,所以需要深入理解这个机制的工作原理。根据论文Gated Axial-Attention结合了轴向注意力和门控机制,轴向注意力指的是在图像的两个轴上(比如行和列)分别应用自注意力,这样可以减少计算量,同时捕捉长程依赖。门控机制则可能用于调整注意力权重,增强重要特征,抑制不相关的部分。 接下来,我需要考虑如何将这个机制转化为代码实现。可能需要参考论文中的结构图和公式。例如,轴向注意力可能分为行注意力和列注意力两个部分,然后通过门控机制进行融合。门控可能使用sigmoid函数来生成0到1之间的权重,然后将这两个注意力输出加权相加。 然后,我需要检查用户提供的代码链接,确保代码结构正确。用户给出的GitHub仓库是jeya-maria-jose/Medical-Transformer,所以应该参考其中的实现。例如,可能在axial_attention.py或者类似的文件中找到Gated Axial-Attention的代码。 此外,需要注意医学图像的特点,比如高分辨率和局部细节的重要性,这可能影响模型的设计。例如,轴向注意力可以处理二维结构,而门控机制帮助模型专注于关键区域,这对分割任务很重要。 用户可能需要具体的代码示例,所以需要提供PyTorch的实现片段,包括轴向注意力的分解、位置编码的添加,以及门控机制的应用。同时,要解释每个部分的作用,比如如何计算查询、键、值,如何应用位置编码,如何生成门控权重,并融合行和列的注意力结果。 还需要考虑实现中的细节,比如如何将二维图像转换为序列以适应Transformer结构,如何处理位置信息,以及如何将轴向注意力模块集成到UNet等分割网络中。可能还需要讨论数据预处理、训练策略和损失函数的选择,比如结合Dice损失和交叉熵损失。 最后,用户可能对如何应用该模型到自己的医学图像数据感兴趣,所以需要给出训练和推理的步骤,比如加载数据、构建模型、设置优化器和训练循环等。同时,提供相关的参考文献和代码库链接,方便用户进一步查阅。</think>以下是为您整理的Medical Transformer中Gated Axial-Attention实现详解: ### 一、Gated Axial-Attention机制解析 1. **轴向注意力分解**:将2D自注意力分解为行、列两个1D注意力操作 $$ \text{Attention}(Q,K,V) = \text{Softmax}(\frac{QK^T}{\sqrt{d_k}} + P)V $$ 其中位置编码$P$分解为$P_{row}$和$P_{col}$[^1] 2. **门控机制实现**: ```python class GatedAxialAttention(nn.Module): def __init__(self, dim, heads): super().__init__() self.row_att = AxialAttention(dim, heads, axis=0) # 行注意力 self.col_att = AxialAttention(dim, heads, axis=1) # 列注意力 self.gate = nn.Parameter(torch.ones(2)) # 可学习门控参数 def forward(self, x): row_out = self.row_att(x) col_out = self.col_att(x) # 门控融合 return torch.sigmoid(self.gate[0])*row_out + torch.sigmoid(self.gate[1])*col_out ``` ### 二、完整实现步骤 1. **特征序列化**:将2D医学图像转换为序列 ```python b, c, h, w = x.shape x_seq = x.view(b, c, h*w).permute(0, 2, 1) # [b, h*w, c] ``` 2. **位置编码注入**: ```python class AxialPositionalEmbedding(nn.Module): def __init__(self, dim, shape): super().__init__() self.row_emb = nn.Parameter(torch.randn(shape[0], dim//2)) self.col_emb = nn.Parameter(torch.randn(shape[1], dim//2)) def forward(self, x): return x + torch.cat([self.row_emb.unsqueeze(1), self.col_emb.unsqueeze(0)], dim=-1).flatten(0,1) ``` 3. **完整网络架构**(基于UNet框架): ```python class MedicalTransformer(nn.Module): def __init__(self, in_ch=3, classes=2): super().__init__() self.encoder = nn.Sequential( ConvBlock(in_ch, 64), GatedAxialAttention(64, heads=4), Downsample(64, 128), GatedAxialAttention(128, heads=8), Downsample(128, 256) ) self.decoder = ... # 对称上采样结构 ``` ### 三、训练实践要点 1. **数据预处理**: - 医学图像标准化(CT值截断+归一化) - 随机旋转/翻转增强 - 器官特定ROI裁剪 2. **损失函数组合**: ```python loss = 0.5*DiceLoss()(pred, mask) + 0.5*CrossEntropyLoss()(pred, mask) ``` 3. **训练策略**: ```python optimizer = AdamW(model.parameters(), lr=1e-4, weight_decay=1e-5) scheduler = CosineAnnealingLR(optimizer, T_max=100) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

勤劳的凌菲

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值