Closed MachineLearningCuiFan closed 2 years ago
同问。
是一个独立的纠错模型,网络结构变化了,参考的softmaskedbert模型,具有一个错误检测网络模块,一个错误纠正网络,loss是两个网络损失的加权。
@shibing624 谢谢回复。
你上传到huggingface的模型应该不包括错误检测网络吧?那么,在huggingface transformers里load_pretrain的时候,如何保证错误检测网络模块的参数也被加载了呢?
换句话说就是我不理解以下这个脚本里哪里包含了错误检测网络模块。
https://github.com/shibing624/pycorrector/blob/master/examples/macbert_demo.py
多谢
错误检测网络的权重被加载了的,模型纠错的流程是先检测再纠正,只是这个过程是端到端的。
PS:pytorch_model.bin权重文件里面detection权重是可以单独看到。
非常感谢!
所以说pycorrector里面是可以直接训练softmasked macbert吗?我看macbert训练的readme里面提到了另一个repo可以用来训练softmasked bert,但好像没有提到pycorrector可不可以。
可以训练softmaskedbert,有abtion写的训练代码。
方便讲解一下detection的权重是怎么糅合到pytorch_model.bin里的呢?换句话说,detection本来是用的另一个网络的权重,怎么转换成transformers包可以直接加载的权重呢?
我仔细阅读了代码,但还是不太理解end-to-end是怎么做到的。我下载了 macbert4csc-base-chinese里的pytorch_model.bin也没看到里面有detection的权重。或者说,你现在huggingface上的其实不包含softmasked bert?
谢谢!
1、huggingface上的模型叫macbert4csc,魔改自softmaskedbert模型,去掉了softmasked结构,detection是一个Linear层,correction是bert的MLM层;训练时loss是detection和correction的加权和,而预测时拿到bert_outputs就可以输出纠错结果,具体参考代码 https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/macbert4csc.py 。 2、字节paper的softmaskedbert模型的实现代码是https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/softmaskedbert4csc.py ,我自评效果比macbert4csc模型差,并未上传到huggingface,大家可以自行评测对比。 3、macbert4csc模型的ckpt模型文件有保存detection权重,pytorch_model.bin没有放detection权重,可以查阅pytorch_model.bin的保存代码https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/train.py#L119 ,主要原因是1)兼容transformers库的调用逻辑;2)不放detection权重也不影响纠错预测结果。
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.(由于长期不活动,机器人自动关闭此问题,如果需要欢迎提问)
1、huggingface上的模型叫macbert4csc,魔改自softmaskedbert模型,去掉了softmasked结构,detection是一个Linear层,correction是bert的MLM层;训练时loss是detection和correction的加权和,而预测时拿到bert_outputs就可以输出纠错结果,具体参考代码 https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/macbert4csc.py 。 2、字节paper的softmaskedbert模型的实现代码是https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/softmaskedbert4csc.py ,我自评效果比macbert4csc模型差,并未上传到huggingface,大家可以自行评测对比。 3、macbert4csc模型的ckpt模型文件有保存detection权重,pytorch_model.bin没有放detection权重,可以查阅pytorch_model.bin的保存代码https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/train.py#L119 ,主要原因是1)兼容transformers库的调用逻辑;2)不放detection权重也不影响纠错预测结果。
作者大大,我在另一个issue里面做过类似提问,在这好像找到了原因,如何确定pytorch_model.bin不放detection权重不影响纠错结果呢?就我个人finetune的结果来看,ckpt模型和bin模型输出差距很大(ckpt模型可以完成新数据的纠错,但bin模型不能),原始的ckpt模型拟合了训练集,但bin模型好像并未拟合,这是否跟bin模型未放detection权重有关呢?
如果是这样的话,需要更新下bin模型的产出逻辑。我再验证下
万分感谢,辛苦!!
想问一下,这个模型是重新根据csc任务特点重新训练的bert吗?----即是利用混淆字词去替换文本产生噪音? 还是本身就是macbert,只是经过那个wang27k数据集微调的模型?