Locally Typical Sampling

rizzearch throwback

мы уже упоминали неидальность maximum likelihood метода вот здесь: задача состоит в максимизации самого правдоподобного из трейна токена, а что будет с остальными - неважно. отчего могут образовываться длинные хвосты в распределении токенов

почему это может быть проблемой? при семплировании всегда есть ненулевая вероятность получить рандомный токен вместо более-менее приемлемого. особенно это остро встает, когда распределение становится остроконечным, и рандомные токены так же острее мешают инференсу

но это малая и не особо вероятная беда. авторы поднимают проблему информативности токенов при их генерации и связыают это с психолингвистами + information theory

в чем идея? как оказывается, самый правдоподобный токен (в терминах likelihood’a) не может быть самым информативным относительно данного контекста, ибо он самый правдоподобный. люди, вроде бы, генерируют свою речь по-другому, стараясь придавать новую информацию каждый раз, при этом сохраняя приверженность контексту

на языке математики это значит что каждый токен находится недалеко относительно ожидаемой энтропии всей последовательности, а значит каждый токен желательно делать не самым вероятным (и не самым невероятным)

как они это реализуют? просто обрубают пик распределения над логитами + обрубают хвосты (при помощи трешхолда от 0 до 1) ⇒ получаем информативно разнообразную адекватную последовательность

тестировались на абстрактной суммаризации и стори генерации, в принципе получилось неплохо. но пригодится такое конечно далеко не во всех задачах

еще интересно подумать над тем, как авторы сдвигают парадигму мышления относительно перплексии: надо не минимизировать ее, а стараться сделать такой же, какая свойственна человеку

👀LINK

Dynamic Stochastic Decoding Strategy for Open-Domain Dialogue Generation

авторы из китая придумали адаптивную температуру. все.

а если поподробнее - авторы реализуют отдельную голову, котора выдает скор разнообразности генерации, и этот скор потом преобразуется в температуру для основного семплинга. преобразований может быть несколько - линейное, экспоненциальное и инверс сигмоида

обучали голову через берт, который показывает похожесть последовательностей → из степени похожести можно вывести степень непохожести и обучать

при том рассказывают, что можно обучать в разных сетапах (фризить основную модель или обучать все вместе), при этом саму адаптивную температуру использовать уже на этапе трейна. выглядит довольно сомнительно, ибо если нигде не врубать стопградиент оператор, то может сильно так перехерачить сходимость этой самой головы (а то и всей модели, если не фризим веса). но кода нет и уточнений по поводу стопградиента тоже, можеть быть это моя паранойя, кто знает

бенчмаркали (если это можно так назвать, потому что датасет они частично собирали сами, частично еще что-то, но в открытый доступ выкладывать не будут) только на диалоговой таске, но можно применять конечно и в других задачах, где нужны разнообразные ответы ллм

и это все на китайских данных

👀LINK

WARP: On the Benefits of Weight Averaged Rewarded Policies

так-так, что-то тут опять дипмаинд намудрили в алайменте

в ходе их следственно-розыскных мероприятий было выявлено, что во время алаймента может наблюдаться

- хакинг ревард модели, которая никогда не может быть идеальной
- катастрофическое забывание из-за ограниченности алаймент датасета (а порой и сфт датасета)
- ну и это выливается в общем счете в потерю вариативности генераций

что же делать? как нам быть? ну, эти авторы думают, что раз доп регуляризации и лоссы не особо помогают, то можно попробовать напрямую мержить веса по образу и подобию ЕМА. но оказалось не все так просто и они используют аж 3 разных способа, которые работают последовательно

- вместо того, чтобы по классике накидывать кл лосс между сфт весами и рлхф весами, тут в качестве якоря используется ЕМА самой полиси, чтобы была и стабилизация, и постепенный апдейт
- далее идет обучение аж нескольких полиси (М штук), которые затем интерполируют между собой и между сфт весами. при том интерполяция не просто линейная, а сферическая относительно сфт весов. почему? да скорее всего потому что с линейной просто не завелось, а дожать до результатов хотелось
- ну и в конце итерации меняем инит модель относительно которая выступает сфт якорем в начале обучения, чтобы обновление весов происходило еще плавнее (ну или возможно мы снова делаем ЕМА на полиси веса, я sooka запутался пока читал их статью и перепутанные обозначения в псевдокоде)

вообще выглядит немного костыльно, мол

давайте попробуем ЕМА - не заработало. окей, давайте тогда сверху накинем еще какой-то мерж весов. и здеся не оч, ну тогда не забудем в финальный аутпут приправить то, с чего начинали - сфт веса. о - вот теперь норм

с другой стороны, может в этом и суть папиры - не всегда стоит вдаваться в теорию, а порой вот такая итеративная долбежка может улучшить скор (хоть и не на долгосрок)

👀LINK

Stochastic Q-learning for Large Discrete Action Spaces

есть классический рл, который на самом деле очень даже интересный и полезный: Q-learning, который в стандартном сетапе создает табличку num_states x num_actions и производит обновление по правилу Беллмана и тд и тп

но что делать если num_actions → inf? (случай, когда num_states → inf можно решать чуть по-другому)

перебирать не по всему пространству, конечно же. но как? и вот авторы подумали как, и теоретически выявили, что на каждом шаге достаточно семплить всего лишь порядка log[num_actions] с округлением вверх, и обновление q-learning’a осуществлять только по этим действиям. при том можно семплить чуть меньше, и хранить вместе с этим буффер уже используемых действий (которые хранятся в связи с exploration соображениями). и, по теории, этого достаточно! супер пупер найс, казалось бы, если не принимать во внимание тот факт, что тестились они на классических средах где не оч много действий (например, 256), когда такая идея с лог сайз буффером и семплингом имеет место быть с количества действий начиная от 10к например.

еще кстати развивают эту идею не только на табулярку, но и на DQN etc. и вроде бы тоже works fine. к тому же можно так же побаловаться с непрерывнями действиями - дискретизировать их и применить этот стохастичный метод, тогда у них в экспах количество действий становилось 4096, что уже становится не оч для рл алгосов, а потому метод (возможно) имеет место быть

смущает только то, что все обучение делали на цпу (немного кринжа в ленту), да и код был бы очень кстати, что именно они запускали и как именно реализовали память и момент семплинга

да и сделать бы им какое-то расширение сред с дейстительно огромным количеством действий, где такое явно необходимо

👀LINK